现在从Nvidia官网下载CUDA
,最新是11.6版本;
然而Pytorch
官网上最新的是pytorch1.11.0+CUDA11.3
;
所以如果使用CUDA-11.6
和pytorch1.11.0+CUDA11.3
,可以正常使用吗?
另外,从Nvidia官网下载的CUDA
和conda下载的CUDA
conda install cuda -c nvidia
有什么区别呢?
可以看这个表
有帮助的 话请采纳一下,谢谢
最好先看一下你的显卡所支持的cuda版本,最新版并不一定是最好的。另外cuda的版本和pytorch的版本也是严格一 一对应的
安装完毕可以用下面的代码测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())