解決PyTorch與CUDA版本不匹配的問(wèn)題
1.CUDA驅(qū)動(dòng)和CUDA Toolkit對(duì)應(yīng)版本
表一:CUDA驅(qū)動(dòng)及CUDA Toolkit最高對(duì)應(yīng)版本
最新可查閱官方文檔
注:驅(qū)動(dòng)是向下兼容的,其決定了可安裝的CUDA Toolkit的最高版本。
2.CUDA Toolkit版本及其可用PyTorch對(duì)應(yīng)版本(參考官網(wǎng))
表二:CUDA Toolkit版本及可用PyTorch對(duì)應(yīng)關(guān)系
CUDAToolkit版本 | 可用PyTorch版本 |
---|---|
7.5 | 0.4.1 ,0.3.0, 0.2.0,0.1.12-0.1.6 |
8.0 | 1.1.0,1.0.0 ,0.4.1 |
9.0 | 1.1.0,1.0.1, 1.0.0,0.4.1 |
9.2 | 1.7.1,1.7.0,1.6.0,1.5.1,1.5.0,1.4.0,1.2.0,0.4.1 |
10.0 | 1.2.0,1.1.0,1.0.1 ,1.0.0 |
10.1 | 1.7.1,1.7.0,1.6.0,1.5.1,1.5.0, 1.4.0,1.3.0 |
10.2 | 1.7.1,1.7.0,1.6.0,1.5.1,1.5.0 |
11.0 | 1.7.1,1.7.0 |
11.1 | 1.8.0 |
注:雖有的卡驅(qū)動(dòng)更新至較新版本,且CUDA Toolkit及PyTorch也可對(duì)應(yīng)更新至新版本。但有的對(duì)應(yīng)安裝包無(wú)法使用,有可能是由于卡太舊的原因。
3.安裝指導(dǎo)
在安裝時(shí)會(huì)同時(shí)安裝CUDA Toolkit以及PyTorch,這是我們要知道的。
步驟一:
使用nvidia-smi查詢驅(qū)動(dòng)版本:
如圖中Driver Version所示,該卡目前的驅(qū)動(dòng)版本為384.81。
步驟二:
此處提供三種方法可供選擇。
(1)指定CUDA Toolkit版本
根據(jù)表一查詢到可安裝的CUDA Toolkit版本,384.81對(duì)應(yīng)最高的CUDA Toolkit版本為9.0。
運(yùn)行conda install pytorch cudatoolkit=9.0 -c pytorch即可。
此方法指定CUDA Toolkit版本后,conda會(huì)自動(dòng)匹配到合適版本的PyTorch。
(2)指定PyTorch版本
根據(jù)表一查詢到可安裝的CUDA Toolkit版本,再根據(jù)表二查詢到合適版本的PyTorch。384.81對(duì)應(yīng)最高的CUDA Toolkit版本為9.0,9.0可安裝PyTorch1.1.0版本。
運(yùn)行conda install pytorch=1.1.0 -c pytorch即可。
此方法指定PyTorch版本后,conda會(huì)自動(dòng)匹配到合適版本的CUDA Toolkit。
(3)同時(shí)指定CUDA Toolkit版本和PyTorch(推薦)
根據(jù)表一查詢到可安裝的CUDA Toolkit版本,根據(jù)表二查詢到合適版本的PyTorch。
運(yùn)行conda install pytorch=1.1.0 cudatoolkit=9.0 -c pytorch即可。
注:PyTorch1.8.0和1.0.0以前版本使用conda安裝時(shí)命令有些許不同,具體可查看官網(wǎng)。
4.驗(yàn)證安裝是否成功
#使用python運(yùn)行 import torch print(torch.__version__) print(torch.cuda.is_available())
卸載當(dāng)前版本PyTorch:
conda uninstall pytorch
補(bǔ)充:查看PyTorch的版本及CUDA和cuDNN版本
檢查PyTorch版本
torch.version # PyTorch version torch.version.cuda # Corresponding CUDA version torch.backends.cudnn.version() # Corresponding cuDNN version torch.cuda.get_device_name(0) # GPU type
更新PyTorch
conda update pytorch torchvision -c pytorch
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
- 淺談pytorch、cuda、python的版本對(duì)齊問(wèn)題
- 粗暴解決CUDA out of memory的問(wèn)題
- 安裝pytorch報(bào)錯(cuò)torch.cuda.is_available()=false問(wèn)題的解決過(guò)程
- vs2022重新編譯opencv-python?cuda加速時(shí)報(bào)錯(cuò)的問(wèn)題解決
- Linux安裝CUDA GCC版本不兼容問(wèn)題
- VScode+cuda編程常見(jiàn)環(huán)境問(wèn)題的解決
- Ubuntu下安裝CUDA10.0以及問(wèn)題
- numba CUDA報(bào)錯(cuò)的問(wèn)題解決
- CUDA環(huán)境下libcuda.so缺失問(wèn)題解決方案
相關(guān)文章
Python?Transformers庫(kù)(NLP處理庫(kù))案例代碼講解
本文介紹transformers 庫(kù)的全面講解,包含基礎(chǔ)知識(shí)、高級(jí)用法、案例代碼及學(xué)習(xí)路徑,內(nèi)容經(jīng)過(guò)組織,適合不同階段的學(xué)習(xí)者,對(duì)Python?Transformers庫(kù)相關(guān)知識(shí)感興趣的朋友一起看看吧2025-04-04python web框架Flask實(shí)現(xiàn)圖形驗(yàn)證碼及驗(yàn)證碼的動(dòng)態(tài)刷新實(shí)例
在本篇文章里小編給大家整理的是關(guān)于python web框架Flask實(shí)現(xiàn)圖形驗(yàn)證碼的相關(guān)知識(shí)點(diǎn),有需要的朋友們參考下。2019-10-10python中count函數(shù)簡(jiǎn)單的實(shí)例講解
在本篇文章里小編給大家整理了關(guān)于python中count函數(shù)簡(jiǎn)單的實(shí)例講解,有興趣的朋友們可以參考下。2020-02-02Python+matplotlib實(shí)現(xiàn)堆疊圖的繪制
Matplotlib作為Python的2D繪圖庫(kù),它以各種硬拷貝格式和跨平臺(tái)的交互式環(huán)境生成出版質(zhì)量級(jí)別的圖形。本文將利用Matplotlib庫(kù)繪制堆疊圖,感興趣的可以了解一下2022-03-03python語(yǔ)言是免費(fèi)還是收費(fèi)的?
在本篇文章里小編給大家分享的是關(guān)于python語(yǔ)言是否免費(fèi)的相關(guān)知識(shí)點(diǎn),需要的朋友們可以學(xué)習(xí)下。2020-06-06OpenCV實(shí)現(xiàn)對(duì)象跟蹤的方法
OpenCV 是一個(gè)很好的處理圖像和視頻的工具,本文主要介紹了OpenCV 進(jìn)行對(duì)象跟蹤,文中通過(guò)示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2021-10-10matplotlib如何設(shè)置坐標(biāo)軸刻度的個(gè)數(shù)及標(biāo)簽的方法總結(jié)
這里介紹兩種設(shè)置坐標(biāo)軸刻度的方法,一種是利用pyplot提交的api去進(jìn)行設(shè)置,另一種是通過(guò)調(diào)用面向?qū)ο蟮腶pi, 即通過(guò)matplotlib.axes.Axes去設(shè)置,需要的朋友可以參考下2021-06-06