安噢 2023-07-04 16:30 采纳率: 75%
浏览 23
已结题

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

  • 写回答

2条回答 默认 最新

  • 你好-C嘉嘉 2023-07-04 16:44
    关注

    有可能你下载的是CPU版本的Pytorch,卸载重装,确保下载的是GPU版本

    img

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 7月26日
  • 已采纳回答 7月18日
  • 创建了问题 7月4日