pytorch安装问题
按照
https://www.bilibili.com/video/BV1hE411t7RN?p=1&vd_source=80205040b90bc67baac793a182ec1007
中的流程安装pytorch,anaconda安装了,然后安装pytorch,在python中import torch是可以的,不会报错,但是用torch.cuda.is_available()时总是返回false,已经换过几个版本的n卡驱动了
(pytorch) C:\Users\admin>python
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False
>>>
n卡信息
C:\Users\admin>nvidia-smi
Tue Jul 4 16:24:38 2023
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 511.09 Driver Version: 511.09 CUDA Version: 11.6 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... WDDM | 00000000:01:00.0 On | N/A |
| N/A 60C P3 29W / N/A | 377MiB / 8192MiB | 1% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1520 C+G N/A |
| 0 N/A N/A 8012 C+G ...artMenuExperienceHost.exe N/A |
| 0 N/A N/A 8532 C+G C:\Windows\explorer.exe N/A |
| 0 N/A N/A 9344 C+G N/A |
| 0 N/A N/A 9896 C+G ...8bbwe\WindowsTerminal.exe N/A |
| 0 N/A N/A 9996 C+G ...n1h2txyewy\SearchHost.exe N/A |
| 0 N/A N/A 10972 C+G ...8bbwe\WindowsTerminal.exe N/A |
| 0 N/A N/A 11620 C+G ...me\Application\chrome.exe N/A |
| 0 N/A N/A 12228 C+G ...2txyewy\TextInputHost.exe N/A |
+-----------------------------------------------------------------------------+
我也不知道安装的是pytorch哪个版本,在官网11.7和11.8的cuda和pip都尝试过,因为刚开始安装会断线,然后换成了清华的源
>>> print(torch.__version__)
1.10.2
>>>
有可能你下载的是CPU版本的Pytorch,卸载重装,确保下载的是GPU版本
对应的pytorch版本是1.7.1的安装命令:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch
我在安装这一步还是很顺利,下载速度也很快,安装完成以后
但是torch.cuda.is_available()依然返回False
看到一个大佬提到他使用国内镜像安装的包全是CPU版本。
我没用国内镜像,但是试着查了一下,竟然真的是cpu版本。
conda list