在学习yolov5时测试

torch 1.8.0 torchvision 0.9.0 cudnn 7.6.5 cudatoolkit 10.2.89

可以使用

torch 和torchvision 可以下

​​https://download.pytorch.org/whl/torch_stable.html​​

寻找下载，然后再pip install ------------

pip install cudnn == 7.6.5, cudatoolkit == 10.2.89 时不行

可以使用conda install cudnn == 7.6.5安装成功

最简单的是这样

conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2