前提:安装后Anaconda和NVIDIA显卡驱动
1. 确定支持的最高CUDA版本
确定支持的最高CUDA版本
记住这个CUDA版本,后面要用。
2. 新建conda虚拟环境
2.1 为conda配置清华镜像源
conda常用命令
打开Anaconda prompt窗口,使用如下命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
2.2 新建指定python版本的conda虚拟环境
新建python 3.9版本的conda虚拟环的境命令:
conda create -n pytorch python=3.9
切换到新建的conda虚拟环境
conda activate pytorch
3. 在conda虚拟环境安装pytorch
3.1 确定要安装的pytorch版本
python和pytorch版本对应关系
举个例子,我选的python版本3.9,最高支持cuda版本为12.2:
torch 1.11及以上都是可以的
这里我选择torch 1.12.1版本
pytorch版本和cuda版本的对应关系
在这个页面找到对应torch 1.12.1版本,根据cuda版本(不能超过第1节中的cuda最高版本)复制对应的命令,在Anaconda prompt中执行:
执行结果如图,则表示成功;如果下载速度很慢,可以更换国内镜像源后,重新执行。
使用如下命令查看刚才新建的conda虚拟环境已安装的包
conda list
4. 测试是否安装成功
依次执行如下命令:
python
import torch
torch.cuda.is_available()
如果输出为True表示安装pytorch成功