日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

解決安裝torch后,torch.cuda.is_available()結果為false的問題_python

作者:l8947943 ? 更新時間: 2023-01-13 編程語言

1. 在conda虛擬環境中安裝了torch

一般命令都可以正常使用,但是使用cuda的命令torch.cuda.is_available()則輸出False。

2. 經過一番查閱資料后

該問題的根本原因是CUDA環境與Torch版本不匹配,因此最直接的解決方式就是使用官方推薦的版本進行適配。

3. 解決思路

查看本機安裝的cuda版本,通過查閱有三個命令

cat /usr/local/cuda/version.txt

nvidia-smi

nvcc -V

有時候會不幸輸出三個不同的版本,這里在虛擬環境下一定以nvcc -V查到的版本為主,作為CUDA版本信息。例如我的版本是10.1

上官網查看對應的安裝命令,然后替換掉現有的版本即可,如下

https://pytorch.org/

找到自己CUDA對應的10.1版本

在虛擬環境中進行安裝

Done,開始測試

4. 其他解決思路

有些人可能只是cudatoolkit版本對不上,如果運氣好,只針對cudatoolkit進行版本匹配即可完成,而且安裝速度還很快安裝命令,記得添加清華源,否則龜速。。。

cudatoolkit=9.0表示你需要對應上的版本

conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

5. 總結

提供了兩種解決torch.cuda.is_available()結果為false的問題

  • 第一種方法是一種萬金油方法比較推薦,但是安裝速度比較感人
  • 第二種方法是精準修復,但是需要自己查好對應的版本

原文鏈接:https://blog.csdn.net/l8947943/article/details/111591610

欄目分類
最近更新