哪個命令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是基于哪個版本構建的。
添加回答
舉報
0/150
提交
取消