pytorch CUDA 版本与 Nvidia CUDA 版本

Question

截至2022年4月26日，CUDA更新至11.6版本，可通过Nvidia Instruction安装：

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（稳定）

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

应该没问题。否则，我看到了here that you can build it from the source (I have python=3.8.13) build instructions

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

pytorch CUDA 版本与 Nvidia CUDA 版本

pytorch CUDA version vs. Nvidia CUDA version

python

cuda

conda

pytorch