亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

Pytorch檢測CUDA

Pytorch檢測CUDA

料青山看我應如是 2023-09-12 17:51:23
哪個命令pytorch可以查看conda env 中看到的 “正確”CUDA 版本?這是一個類似的問題,但并沒有讓我走得太遠。nvidia-smi說我有cuda版本10.1conda list告訴我 cudatoolkit 版本是10.2.89torch.cuda.is_available()顯示 FALSE,所以它看到No CUDA?print(torch.cuda.current_device()),我得到10.0.10(10010??)(看起來像):AssertionError:您系統上的 NVIDIA 驅動程序太舊(發現版本 10010)print(torch._C._cuda_getCompiledVersion(), 'cuda compiled version')告訴我我的版本是10.0.20(10020??)?10020 cuda編譯版本為什么有這么多不同的版本?我缺少什么?PS 我在 Ubuntu 16.04 和 Geforce 1050 上有 Nvidia 驅動程序。當我從ubuntu ( ) 中的選項卡安裝驅動程序時,430它附帶了。我安裝了 其中還安裝了使用libcuda1-430additional driversSoftware and Updatespytorchcondacudatoolkitconda install -c fastai -c pytorch -c anaconda fastai
查看完整描述

1 回答

?
紅顏莎娜

TA貢獻1842條經驗 獲得超13個贊

在安裝 pytorch 的 conda env (myenv) 中執行以下操作:

conda activate myenv

torch.version.cuda

Nvidia-smi僅顯示兼容版本。好像沒有講到pytorch自己的cuda是基于哪個版本構建的。


查看完整回答
反對 回復 2023-09-12
  • 1 回答
  • 0 關注
  • 132 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號