一、下载Anaconda
1.由于anaconda的服务器都在国外,推荐大家使用镜像源进行下载,清华的conda镜像链接: anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirrora
2.使用命令新建一个虚拟环境,我这里将环境名称写为pytorch:
conda create -n pytorch python==3.11.5
(格式为:conda create -n “你想要的名字” python== “某python版本”)
3.切换到该环境
conda activate pytorch
二、安装CUDA、cuDNN
1.打开cmd输入 nvidia-smi 命令查看驱动版本与cuda版本 :
参考pytorch官网,更新cuda
cuda11.8下载地址如下:
https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
命令号执行nvcc -V
三、安装pytorch
按照官网,执行如下指令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
四、验证是否安装成功
python
import torch
print(torch.__version__)
print(torch.version.cuda)
torch.cuda.is_available()
print(torch.cuda.device_count())
输出pytorch版本、True即表示安装成功!