我的cuda版本显示在这里。 nvcc -V 输出 nvidia-smi 输出
我基本上想安装apex。我首先使用命令
conda install pytorch=1.10.1 cudatoolkit=11.1 -c pytorch
用cuda安装torch，这个版本的cudatoolkit工作正常并且
torch.version.cuda
正确显示预期输出“11.1”并且
torch.cuda.is_available()
返回 True。然而，当我稍后尝试安装 apex 时，这不起作用，因为我的 nvcc -V 的版本如上所示是 11.4，但 cudatoolkit 是 11.1。
然后我使用命令
conda install pytorch cudatoolkit=11.4 -c pytorch
安装正确的 cudatoolkit 版本 11.4。这样做之后，torch版本变为1.9.1。然而，
torch.version.cuda
变为无并且
torch.cuda.is_available()
返回 False。
我还尝试了 cudatoolkit=11.4 和 torch 版本 1.9.1 和 1.12.1，但它们都不起作用。
我希望我的 GPU 能够被正确检测到。我应该使用哪个版本的 torch 和 cudatoolkit？或者这里还有什么问题吗？多谢。