潇洒郎: Cuda9.0 100%安装pytorch 成功

显卡驱动版本决定cuda版本,cuda版本决定cudnn版本与pytorch版本

驱动版本390.65,——cuda90——cudnn90——pytorch

版本有一个对不上,torch.cuda.is_available() 都是False

Cuda9.0 搭配 pytorch1.1.0 torchvision0.3.0 torchaudio不用

下载whl文件直接pip install 安装即可

 

1、查看自己显卡驱动版本

潇洒郎: Cuda9.0 100%安装pytorch 成功_第1张图片

C:\Program Files\NVIDIA Corporation\NVSMI>nvidia-smi

NVIDIA-SMI 390.65  DRIVER VERSION:390.6

你可能感兴趣的:(Python学习)