pytorch CUDA 版本与 Nvidia CUDA 版本

Question

截止2022年4月26日，CUDA已更新至11.6版本，可通过英伟达安装说明：

wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
sudo sh cuda_11.6.2_510.47.03_linux.run

我猜cudatoolkit的版本也会是11.6

但是，没有与pytorch匹配的CUDA11.6版本。

在pytorch的网站上，最新的CUDA版本是11.3 ， pytorch版本将是1.11.0(stable)

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

因此，如果我将CUDA11.6和pytorch1.11.0与cudatoolkit=11.3一起使用，它会正常执行吗？ 如果下面的Nvidia Instruction和conda method有什么区别？

conda install cuda -c nvidia

此致！

Answer 1

应该没问题。 否则，我在这里看到您可以从源代码构建它（我有 python=3.8.13）构建指令

pip install torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116