在学习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