pytorch安装错误,torch.cuda.is_available()返回false

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版本

img

  • 这个问题的回答你可以参考下: https://ask.csdn.net/questions/7449661
  • 这篇博客也不错, 你可以看下安装pytorch+cuda9.0
  • 除此之外, 这篇博客: pytorch安装,torch.cuda.is_available()返回False解决方案中的 安装对应版本pytorch 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  • pytorch历史版本

    对应的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