环境配置
PATH环境配置:
下载torch和torchvision版本离线下载地址
https://download.pytorch.org/whl/torch_stable.html
在虚拟环境上安装:
pip install C:\Project\torch-1.2.0-cp37-cp37m-win_amd64.whl
pip install C:\Project\torchvision-0.4.0-cp37-cp37m-win_amd64.whl
其他方面:
一、更新GPU驱动
官方驱动 | NVIDIA到官网下载对应的显卡驱动
nvidia-smi命令查看GPU信息
二 、cuda安装
选择对应的cuda版本,注意:cuda版本不能高于,显卡支持的最大版本
下载11.2
安装,每次下一步操作
测试cuda是否安装成功 ,nvcc -V
三、安装cudnn安装
下载:https://developer.nvidia.com/zh-cn/cudnn
Cudnn版本需要和cuda一致
复制cudnn对应目录到cuda安装目录
四、选择pytorch版本
Pytorch: Previous PyTorch Versions | PyTorch
找到对应pytorch版本对应的cudn和cudann命令
测试pytorch_GPU是否可以用
python
import torch
print(torch.is_available())