windows 10安装GPU版本pytorch

一、下载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即表示安装成功！