安裝Pytorch時CUDA的選擇

• • Nvidia CUDA
  • 查看CUDA版本方法
  • • 查看CUDA的驅動API版本
    • 查看CUDA的運行API版本
    • • 查看官方CUDA的運行API版本
      • • 方法1
        • 方法2
      • 查看Anaconda里cudatoolkit包版本

CUDA是一個并行計算平臺和編程模型，能夠使得使用GPU進行通用計算變得簡單和優雅。Nvidia官方提供的CUDA 庫是一個完整的工具安裝包，其中提供了 Nvidia驅動程序、開發 CUDA 程序相關的開發工具包等可供安裝的選項。

Nvidia CUDA

Nvidia安裝的cuda是一個完整的cuda庫，包含cuda toolkit和相應的驅動。主要包含三部分：CUDA toolkit、CUDA driver和NVIDIA GPU driver。理論上在安裝完驅動以后只需要安裝CUDA toolkit就可以保證CUDA相關的程序運行。

查看CUDA版本方法

CUDA分為兩種，驅動API和運行API，驅動API指的是指的顯卡驅動支持的最高cuda版本，我們運行程序時用的是運行API。

查看CUDA的驅動API版本

表示的是驅動所能支持的最大運行API版本。如果顯示11.3，用我們常說的話就是最大支持11.3的cuda。下面為查看CUDA驅動API版本的命令：

nvidia-smi #右上角顯示CUDA版本(驅動API)

顯示結果：

查看CUDA的運行API版本

這是平時我們所說的CUDA版本，由于運行API在CUDA里的CUDA Toolkit工具包中，所以運行API版本也是CUDA Toolkit工具包的版本。其實裝了Anaconda之后Anaconda也會提供一個cudatoolkit工具包，同樣包含了CUDA的運行API，可以用來替代官方CUDA的CUDA Toolkit。這也就是為什么有時候我們通過nvcc-V查看的cuda版本很低(比如7.5)，但是能成功運行cuda9.0的pytorch的原因。因為在安裝完anaconda后，運行pytorch代碼就會使用anaconda的cudatoolkit，而忽視官方的CUDA Toolkit，所以我們只需要根據anaconda的cudaoolkit包的版本來安裝相應的pytorch即可。下面為查看運行API版本的命令：

查看官方CUDA的運行API版本

方法1

nvcc -V

結果顯示官方CUDA提供的CUDA Toolkit工具包的版本為7.5。

方法2

輸入下面的命令，也是顯示官方提供的CUDA Toolkit版本。顯示的版本和上面一樣，也是7.5。

cat /usr/local/cuda/version.txt

查看Anaconda里cudatoolkit包版本

輸入以下命令，在顯示列表中找到cudatoolkit包就可以看到它的版本了。

conda list

顯示的版本為10.2.89。這是我們運行pytorch的cuda版本，根據它來下載對應的pytorch即可。

以上為本人參考了博客和動手實踐得出的結論，如有錯誤可在評論區指出，方便一起學習。

參考博客：
顯卡，顯卡驅動,nvcc, cuda driver,cudatoolkit,cudnn到底是什么？
CUDA、NVIDIA driver、多版本cuda

總結

以上是生活随笔為你收集整理的安装Pytorch如何选择CUDA的版本的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。