安装ANACONDA
ANACONDA点击跳转官网直接下载最新版本
安装CUDA(查版本号即可)
CUDA对应的显卡版本号查询
下载安装pytorch
pytorch 我的CUDA对应版本为
打开Anaconda Prompt切换到自己的环境并复制以下语句
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
输入一下命令验证安装是否成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda_version)
显示版本号则为安装成功!
1.10.2
True
11.3