pytorch 1.7是否正式启用于cuda 10.0?

Question

我的个人项目只能使用CUDA 10.0。

而不是安装Pytorch与版本适合CUDA 10.0，我不小心安装Pytorch 1.7与CUDA 10.1支持。特别地，我通过

安装它

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

令人惊讶的是，尽管CUDA版本不匹配，但到目前为止一切都运行良好。

为了验证我的安装，我运行了collect_env.py中给出的代码，并且它很好。

我只是想知道一些事情。

1. Pytorch团队是否正式评论Pytorch 1.7与CUDA 10.0兼容?
是否有更严格的方法来验证我的Pytorch安装?

Answer 1

令人惊讶的是，尽管CUDA版本不匹配，但到目前为止一切正常。

小版本之间的更改应该可以工作(像这样的不匹配在我的情况下工作)，尽管在10.x版本(源代码)中没有兼容性承诺，只有从11.x开始才会有二进制兼容性。

Pytorch团队是否正式评论Pytorch 1.7与CUDA 10.0兼容?

我不知道，但是列出的轮子不包括10.0CUDA和PyTorch1.7.0(最新的10.0支持似乎是1.4.0)。

是否有更严格的方法来验证我的Pytorch安装?

如上所述，也许可以克隆PyTorch的GitHub repo，恢复到标记的版本，并运行测试(文件夹在这里，cuda测试文件之一在这里)，但对于个人项目可能过多。

Answer 2

Pytorch先前版本日志提供了CUDA 9.2, 10.1, 10.2和11版本的安装。因此，CUDA 10可能不被官方支持。