我今天安裝了 Anaconda、CUDA 和 PyTorch，但我無法在 Torch 中訪問我的 GPU（RTX 2070）。我遵循了所有安裝步驟，否則 PyTorch 作業正常，但是當我嘗試在 shell 或腳本中訪問 GPU 時，我得到
>>> import torch
>>> torch.cuda.is_available()
False
>>> torch.cuda.device_count()
0
>>> print(torch.version.cuda)
None
運行conda list顯示這是我安裝的包
cudatoolkit 11.3.1 h59b6b97_2
并numba -s在 conda 環境中運行顯示
__CUDA Information__
CUDA Device Initialized : True
CUDA Driver Version : 11030
CUDA Detect Output:
Found 1 CUDA devices
id 0 b'NVIDIA GeForce RTX 2070' [SUPPORTED]
compute capability: 7.5
pci device id: 0
pci bus id: 1
Summary:
1/1 devices are supported
并且所有測驗都通過ok。CUDA 11.3 是 PyTorch 和我的 GPU 支持的計算平臺之一，這就是我安裝的版本。
我已經嘗試重新安裝 CUDA，我在 Windows 10 上，nvcc --version顯示已安裝 CUDABuild cuda_11.3.r11.3/compiler.29745058_0
任何的意見都將會有幫助
編輯：我正在使用從他們網站上生成的命令安裝的 PyTorch 1.10 。使用python 3.9.7. 我還在新的 conda 環境中再次安裝了 PyTorch，但遇到了同樣的問題。
uj5u.com熱心網友回復：
我建議重新啟動您的系統并再次檢查它是否正常作業？
Pytorch你安裝的是哪個版本？您是否嘗試在Pytorch和Cuda版本之間進行匹配？
uj5u.com熱心網友回復：
首先，打開一個終端并運行 nvidia-smi 并查看它是否檢測到您的 GPU。如果可行，請仔細檢查您的 cuda 版本是否與 PyTorch 要求的相同。如果您有舊版本的 Cuda，請下載最新版本，重新啟動計算機并重試。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/334324.html
下一篇：元組串列的隨機抽樣