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