[Python知识库]Windows 10 安装cuda、PyTorch

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

?安装完成后测试一下：

1. import torch

2. print(torch.__version__)

3. print(torch.cuda.is_available())

?