torch安装

torch安装;支持cuda

pip install torch==1.8.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

torchvision安装

 pip uninstall torchvision --no-deps;pip install torchvision==0.9

1. 使用--no-deps参数仅卸载目标包
2. 验证cuDNN安装是否成功。可以运行以下代码验证:

python
import torch
torch.backends.cudnn.is_acceptable(torch.tensor(1.0))

如果没有错误输出,则表示cuDNN安装成功。
3. 检查你的CUDA版本是否支持cuDNN。cuDNN需要CUDA 8.0或更高版本。可以运行以下代码检查CUDA版本:

python
import torch
print(torch.version.cuda)`

2. 如果CUDA版本符合要求,安装与CUDA版本对应的cuDNN版本。可以到NVIDIA官网下载对应版本的cuDNN。

3. 检查你的PyTorch版本和torchvision版本是否兼容。可以使用以下代码检查:

python
import torch 
print(torch.__version__)

import torchvision
print(torchvision.__version__) 

5. 对照torchvision安装页面上的兼容性矩阵,确认两者版本是否兼容。

python -c "import torch;print(torch.backends.cudnn.is_acceptable(torch.tensor(1.0)))"
False

python -c "import torch;print(torch.backends.cudnn.enabled)"
True