解決安裝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版本不匹配，因此最直接的解決方式就是使用官方推薦的版本進(jìn)行適配。
3. 解決思路
查看本機(jī)安裝的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)境中進(jìn)行安裝
Done，開始測試
4. 其他解決思路
有些人可能只是cudatoolkit版本對不上，如果運(yùn)氣好，只針對cudatoolkit進(jìn)行版本匹配即可完成，而且安裝速度還很快安裝命令，記得添加清華源，否則龜速。。。
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的問題
- 第一種方法是一種萬金油方法比較推薦，但是安裝速度比較感人
- 第二種方法是精準(zhǔn)修復(fù)，但是需要自己查好對應(yīng)的版本
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了用Python編寫分析Python程序性能的工具教程,包括了對內(nèi)存泄漏等常見問題的檢測方法,需要的朋友可以參考下2015-04-04
Python使用Vagrant搭建開發(fā)環(huán)境的具體步驟
使用 Vagrant 搭建開發(fā)環(huán)境是一個非常方便的方式,它可以幫助你快速創(chuàng)建、配置和管理虛擬機(jī),確保開發(fā)環(huán)境的一致性,以下是使用 Vagrant 搭建開發(fā)環(huán)境的具體步驟,需要的朋友可以參考下2024-09-09
之前的《Python中的隨機(jī)采樣和概率分布我們介紹了Python中最簡單的隨機(jī)采樣函數(shù)。接下來我們更進(jìn)一步，來看看如何從一個概率分布中采樣，我們以幾個機(jī)器學(xué)習(xí)中最常用的概率分布為例。2022-01-01
這篇文章主要介紹了Python中快速掌握Data Frame的常用操作,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-03-03
使用django-guardian實(shí)現(xiàn)django-admin的行級權(quán)限控制的方法
這篇文章主要介紹了使用django-guardian實(shí)現(xiàn)django-admin的行級權(quán)限控制的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-10-10
Python數(shù)據(jù)可視化庫seaborn的使用總結(jié)
這篇文章主要介紹了Python數(shù)據(jù)可視化庫seaborn的使用總結(jié)，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了Python計算公交發(fā)車時間的完整代碼，代碼簡單易懂，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2020-02-02
簡單介紹Python的Tornado框架中的協(xié)程異步實(shí)現(xiàn)原理
這篇文章主要介紹了簡單介紹Python的Tornado框架中的協(xié)程異步實(shí)現(xiàn)原理,作者基于Python的生成器講述了Tornado異步的特點(diǎn),需要的朋友可以參考下2015-04-04
python中浮點(diǎn)數(shù)比較判斷!為什么不能用==(推薦)
這篇文章主要介紹了python中浮點(diǎn)數(shù)比較判斷!為什么不能用==,本文給大家分享問題解決方法,需要的朋友可以參考下2023-09-09
這篇文章主要介紹了基于pycharm 項目和項目文件命名規(guī)則的介紹，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-01-01