安裝Pytorch時(shí)CUDA的選擇

• • Nvidia CUDA
  • 查看CUDA版本方法
  • • 查看CUDA的驅(qū)動(dòng)API版本
    • 查看CUDA的運(yùn)行API版本
    • • 查看官方CUDA的運(yùn)行API版本
      • • 方法1
        • 方法2
      • 查看Anaconda里cudatoolkit包版本

CUDA是一個(gè)并行計(jì)算平臺(tái)和編程模型，能夠使得使用GPU進(jìn)行通用計(jì)算變得簡單和優(yōu)雅。Nvidia官方提供的CUDA 庫是一個(gè)完整的工具安裝包，其中提供了 Nvidia驅(qū)動(dòng)程序、開發(fā) CUDA 程序相關(guān)的開發(fā)工具包等可供安裝的選項(xiàng)。

Nvidia CUDA

Nvidia安裝的cuda是一個(gè)完整的cuda庫，包含cuda toolkit和相應(yīng)的驅(qū)動(dòng)。主要包含三部分：CUDA toolkit、CUDA driver和NVIDIA GPU driver。理論上在安裝完驅(qū)動(dòng)以后只需要安裝CUDA toolkit就可以保證CUDA相關(guān)的程序運(yùn)行。

查看CUDA版本方法

CUDA分為兩種，驅(qū)動(dòng)API和運(yùn)行API，驅(qū)動(dòng)API指的是指的顯卡驅(qū)動(dòng)支持的最高cuda版本，我們運(yùn)行程序時(shí)用的是運(yùn)行API。

查看CUDA的驅(qū)動(dòng)API版本

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

nvidia-smi #右上角顯示CUDA版本(驅(qū)動(dòng)API)

顯示結(jié)果：

查看CUDA的運(yùn)行API版本

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

查看官方CUDA的運(yùn)行API版本

方法1

nvcc -V

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

方法2

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

cat /usr/local/cuda/version.txt

查看Anaconda里cudatoolkit包版本

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

conda list

顯示的版本為10.2.89。這是我們運(yùn)行pytorch的cuda版本，根據(jù)它來下載對(duì)應(yīng)的pytorch即可。

以上為本人參考了博客和動(dòng)手實(shí)踐得出的結(jié)論，如有錯(cuò)誤可在評(píng)論區(qū)指出，方便一起學(xué)習(xí)。

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。