淘先锋技术网

首页 1 2 3 4 5 6 7

Win10下配置Pytorch-GPU（CUDA10.1）

Win10下配置Pytorch-GPU（CUDA10.1）

一、 CUDA安装

官网点击此处，根据需要版本下载安装，这里我选择的是CUDA Toolkit 10.1 update2 (Aug 2019)版本

在这里插入图片描述
在这里插入图片描述
点击Download下载安装包即可，CUDA的安装就一直默认选项安装
在命令行窗口输入

nvcc -V

查验CUDA是否安装成功
在这里插入图片描述
可以看到CUDA的版本号

二、CUDNN下载

官网点击此处下载CUDNN，CUDNN的版本要与CUDA版本保持一致。这里我选择的版本如下：
在这里插入图片描述
下载完成解压缩，将CUDNN里面的bin、clude、lib文件直接复制到CUDA的安装目录，如下图为CUDA的安装位置，粘贴过来直接覆盖即可。
在这里插入图片描述

三、Pytorch-GPU安装

Pytorch官网点击此处，根据配置进行选择。这里我选择pip安装，由于网络问题和其他因素，pip下载很慢，这里做一些修改，从清华镜像下载，然后等待下载直至安装成功，若报错，多试几次就好。
在这里插入图片描述
在这里插入图片描述

pip install --default-timeout=1000 -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

安装过程：
在这里插入图片描述
命令行验证pytorch是否安装成功

import torch
print(torch.__version__)

在这里插入图片描述
接下来验证能否使用gpu

print(torch.cuda.is_available())

在这里插入图片描述
返回True，安装成功！