哪個(gè)命令pytorch可以查看conda env 中看到的 “正確”CUDA 版本?這是一個(gè)類似的問題,但并沒有讓我走得太遠(yuǎn)。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:您系統(tǒng)上的 NVIDIA 驅(qū)動(dòng)程序太舊(發(fā)現(xiàn)版本 10010)print(torch._C._cuda_getCompiledVersion(), 'cuda compiled version')告訴我我的版本是10.0.20(10020??)?10020 cuda編譯版本為什么有這么多不同的版本?我缺少什么?PS 我在 Ubuntu 16.04 和 Geforce 1050 上有 Nvidia 驅(qū)動(dòng)程序。當(dāng)我從ubuntu ( ) 中的選項(xiàng)卡安裝驅(qū)動(dòng)程序時(shí),430它附帶了。我安裝了 其中還安裝了使用libcuda1-430additional driversSoftware and Updatespytorchcondacudatoolkitconda install -c fastai -c pytorch -c anaconda fastai
Pytorch檢測CUDA
料青山看我應(yīng)如是
2023-09-12 17:51:23