網站首頁 編程語言 正文
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
相關推薦
- 2024-01-06 RocketMQ重復消費問題
- 2022-10-06 Redis中鍵和數據庫通用指令詳解_Redis
- 2022-12-24 Kubernetes?ApiServer三大server權限與數據存儲解析_云和虛擬化
- 2023-10-12 v-if和v-for的優先級以及二者同時使用的情況
- 2022-08-01 詳解docker?API管理接口增加CA安全認證_docker
- 2022-11-17 Go語言學習教程之反射的示例詳解_Golang
- 2022-10-26 如何查看git分支從哪個源分支拉的_相關技巧
- 2022-09-16 記一次SQL優化的實戰記錄_數據庫其它
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支