初入CV第一课配置环境Annaconda+Pytorch+CUDA+Cudnn
因为我们需要使用CUDA+Cudnn,因此不能使用conda进行下载,否则会下载cpu版本的torch,然后一直出现torch.cuda.is_available()返回False的问题。因为我的电脑的CUDA版本为11.6,遵循版本向下兼容原则,去Nvidia官网下载11.3版本的CUDA。一定要先查一遍自己的电脑有没有安装旧版的CUDA,有就要卸载干净再重新装新版的CUDA。因为选择的是11.3版本,因此需要将最后的数字改成113。5、查看电脑的CUDA版本并从官网下载CUDA。
原创
2023-07-11 21:25:53 ·
999 阅读 ·
3 评论