1、安装cuda
官网链接:https://developer.nvidia.com/cuda-downloads
根据需要进行选择,这里安装11.3,具体选择见下图,选择本地安装。
下载完成后,双击exe文件。
选择精简安装
打开cmd输入nvcc --version,显示版本号即为安装成功:
cmd中输入set cuda,可以查看 CUDA 设置的环境变量。
CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3
CUDA_PATH_V11_3=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3
2、安装cuDNN
下载地址:https://developer.nvidia.com/rdp/cudnn-download
下载完成后,解压缩。将cuDNN 解压缩后的三个文件夹(bin、include、lib)中的配置文件添加到 CUDA安装目录中的同名文件夹中。
完成添加后,系统环境变量中的 path 添加如下路径(根据自己的路径进行修改)
1.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
2.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp
3.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include
4.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib
3、安装python
当前Windows下的PyTorch仅支持Python 3.7-3.9。这里选择下载Windows installer (64-bit)并安装3.8.10。
链接https://www.python.org/downloads/release/python-3810/
注:修改pip源方法
# 清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
# 阿里源
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
# 腾讯源
pip config set global.index-url http://mirrors.cloud.tencent.com/pypi/simple
# 豆瓣源
pip config set global.index-url http://pypi.douban.com/simple/
这些命令将在C:\Users\zlj\AppData\Roaming\pip下新建一个pip.ini配置文件。
4、安装PyTorch
官网链接:https://pytorch.org/get-started/locally/
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
安装完成后测试一下:
-
import torch
-
print(torch.__version__)
-
print(torch.cuda.is_available())