cuda安装

环境配置

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安装

1. CUDA Toolkit Archive | NVIDIA Developer

选择对应的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())