安装Pytorch如何选择CUDA的版本
安裝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)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 比较丧的网名女82个
- 下一篇: pytorch使用GPU炼丹笔记