第一次安装Pytorch

1、新版本的Anaconda内置的python版本是3.12，
目前 Windows 上的 PyTorch 仅支持 Python 3.8-3.11;不支持 Python 2.x。

在这里插入图片描述

1、创建运行环境

在不创建虚拟环境的情况下，不建议使用最新的Python和Anaconda。
在几次失败后，我使用的是Anaconda3-2023.09-0-Windows-x86_64版本。
在清华源下载：Anaconda3-2023.09-0-Windows-x86_64
在这里插入图片描述
因为下载的是旧版Anaconda，很多自带库都需要更新。不过这都是非必要的。安装Anaconda教程可以参考 **一文安装anaconda **

安装好Anaconda，在开始菜单中找到 Anaconda Prompt，右键单击，点击更多，打开文件位置，将 Jupyter Notebook和Anaconda Prompt复制到桌面上。
在这里插入图片描述
双击打开Anaconda Prompt，输入命令更新pip

python">python -m pip install --upgrade pip

2、安装cuda

可以先查看自己的cuda版本，
1、打开NVIDIA控制面板。
2、点击系统信息。
3、点击组件。查看NVIDIA CUDA 版本。
如图：我安装的是 12.4

在这里插入图片描述
这里建议的CUDA版本是11.8、12.1、12.4

安装CUDA方法：点击进入 CUDA-toolkit-archive

选择合适的版本进行安装。
譬如：11.8、12.1、12.4
在这里插入图片描述

3、安装Pytorch

这个是截止到2024年9月，Pytorch的安装环境需求。
在这里插入图片描述

如果CUDA版本同样是12.4，可以直接运行以下代码：

python">pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

如果CUDA版本是12.1，可以直接运行以下代码：

python">pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

如果CUDA版本同样是11.8，可以直接运行以下代码：

python">pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

打开Anaconda Prompt，输入上述对应版本的命令。
在这里插入图片描述
发现时间太长，可以更换pip源。
我想使用清华源下载，发现还是太慢。
最后使用迅雷下载whl，然后pip下载。
情况如下：
复制上述链接：
https://download.pytorch.org/whl/cu124/torch-2.4.1%2Bcu124-cp311-cp311-win_amd64.whl
打开迅雷，下载。
在Anaconda Prompt中输入

python">pip install "C:\Users\程辉\Downloads\torch-2.4.1+cu124-cp311-cp311-win_amd64.whl"

我下载的Pytorch是在
E:\迅雷下载\torch-2.4.1+cu124-cp311-cp311-win_amd64.whl
如果不想输入位置，可以先输入 pip install 后将文件夹中的torch拖入到Anaconda Prompt中，回车运行。
等待安装完成，会显示：torch is already installed…
可以尝试输入以下代码进行检测
在这里插入图片描述

python">python
import torch
torch.cuda.is_available()

返回True就代表可以安装完成了。