解決安裝torch后,torch.cuda.is_available()結(jié)果為false的問(wèn)題
1. 在conda虛擬環(huán)境中安裝了torch
一般命令都可以正常使用，但是使用cuda的命令torch.cuda.is_available()則輸出False。
2. 經(jīng)過(guò)一番查閱資料后
該問(wèn)題的根本原因是CUDA環(huán)境與Torch版本不匹配，因此最直接的解決方式就是使用官方推薦的版本進(jìn)行適配。
3. 解決思路
查看本機(jī)安裝的cuda版本，通過(guò)查閱有三個(gè)命令
cat /usr/local/cuda/version.txt nvidia-smi nvcc -V
有時(shí)候會(huì)不幸輸出三個(gè)不同的版本，這里在虛擬環(huán)境下一定以nvcc -V查到的版本為主，作為CUDA版本信息。例如我的版本是10.1
上官網(wǎng)查看對(duì)應(yīng)的安裝命令，然后替換掉現(xiàn)有的版本即可，如下
找到自己CUDA對(duì)應(yīng)的10.1版本
在虛擬環(huán)境中進(jìn)行安裝
Done，開(kāi)始測(cè)試
4. 其他解決思路
有些人可能只是cudatoolkit版本對(duì)不上，如果運(yùn)氣好，只針對(duì)cudatoolkit進(jìn)行版本匹配即可完成，而且安裝速度還很快安裝命令，記得添加清華源，否則龜速。。。
cudatoolkit=9.0表示你需要對(duì)應(yīng)上的版本
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
5. 總結(jié)
提供了兩種解決torch.cuda.is_available()結(jié)果為false的問(wèn)題
- 第一種方法是一種萬(wàn)金油方法比較推薦，但是安裝速度比較感人
- 第二種方法是精準(zhǔn)修復(fù)，但是需要自己查好對(duì)應(yīng)的版本
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
PyQt5中QPushButton的用法詳細(xì)解析與應(yīng)用實(shí)戰(zhàn)
PyQt5 是一個(gè)用于創(chuàng)建圖形用戶界面的 Python 綁定庫(kù),它基于 Qt5 應(yīng)用程序框架,在 PyQt5 中,QPushButton 是一個(gè)常用的控件,用于創(chuàng)建按鈕,允許用戶通過(guò)點(diǎn)擊來(lái)觸發(fā)某些操作,本文將詳細(xì)介紹 QPushButton 的用法,并通過(guò)實(shí)際案例來(lái)展示其強(qiáng)大的功能2024-07-07
Python實(shí)現(xiàn)的微信公眾號(hào)群發(fā)圖片與文本消息功能實(shí)例詳解
這篇文章主要介紹了Python實(shí)現(xiàn)的微信公眾號(hào)群發(fā)圖片與文本消息功能,結(jié)合實(shí)例形式詳細(xì)分析了Python調(diào)用微信接口實(shí)現(xiàn)微信公眾號(hào)群發(fā)圖片與文本消息的具體操作步驟與相關(guān)注意事項(xiàng),需要的朋友可以參考下2017-06-06
Python數(shù)據(jù)分析之pandas函數(shù)詳解
這篇文章主要介紹了Python數(shù)據(jù)分析之pandas函數(shù)詳解,文中有非常詳細(xì)的代碼示例,對(duì)正在學(xué)習(xí)python的pandas函數(shù)的小伙伴們有很好地幫助,需要的朋友可以參考下2021-04-04
python軟件測(cè)試Jmeter性能測(cè)試JDBC Request(結(jié)合數(shù)據(jù)庫(kù))的使用詳解
這篇文章主要介紹了python軟件測(cè)試Jmeter性能測(cè)試JDBC Request(結(jié)合數(shù)據(jù)庫(kù))的使用詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
python中自帶的三個(gè)裝飾器的實(shí)現(xiàn)
這篇文章主要介紹了python中自帶的三個(gè)裝飾器的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-11-11
python自動(dòng)化測(cè)試selenium執(zhí)行js腳本實(shí)現(xiàn)示例
這篇文章主要為大家介紹了python自動(dòng)化測(cè)試selenium執(zhí)行js腳本的實(shí)現(xiàn)示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步2021-11-11
這篇文章主要介紹了Python中處理時(shí)間的幾種方法,包括時(shí)間的獲取和時(shí)間之間的轉(zhuǎn)換等等,需要的朋友可以參考下2015-04-04
Python求區(qū)間正整數(shù)內(nèi)所有素?cái)?shù)之和的方法實(shí)例
這篇文章主要給大家介紹了Python對(duì)區(qū)間正整數(shù)內(nèi)所有素?cái)?shù)之和的相關(guān)資料，文中介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10
本文主要介紹了Numpy隨機(jī)抽樣的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
pandas把dataframe轉(zhuǎn)成Series,改變列中值的類(lèi)型方法
下面小編就為大家分享一篇pandas把dataframe轉(zhuǎn)成Series,改變列中值的類(lèi)型方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-04-04