深度学习torch torchvision cudnn cudatoolkit

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

你可能感兴趣的:(深度学习,pytorch,机器学习)