0
点赞
收藏
分享

微信扫一扫

深度学习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



举报

相关推荐

0 条评论