显卡驱动版本决定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、查看自己显卡驱动版本
C:\Program Files\NVIDIA Corporation\NVSMI>nvidia-smi
NVIDIA-SMI 390.65 DRIVER VERSION:390.65