Nvidia CUDA版本和Pytorch CUDA版本兼容性

现在从Nvidia官网下载CUDA,最新是11.6版本;
然而Pytorch官网上最新的是pytorch1.11.0+CUDA11.3
所以如果使用CUDA-11.6pytorch1.11.0+CUDA11.3,可以正常使用吗?
另外,从Nvidia官网下载的CUDA

和conda下载的CUDA

conda install cuda -c nvidia

有什么区别呢?

可以看这个表

img

有帮助的 话请采纳一下,谢谢

最好先看一下你的显卡所支持的cuda版本,最新版并不一定是最好的。另外cuda的版本和pytorch的版本也是严格一 一对应的
安装完毕可以用下面的代码测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())