视频链接:PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】_哔哩哔哩_bilibili

在我学习改视频的第一章就有点问题就是torch.cuda.is_available(),false ,我寻思我明明有显卡咋会false,然后我就自己开始鼓捣了。

问题一:cuda不适配?

首先win+r,输入cmd

nvidia-smi

可以弹出如下框所示回应

Mon Dec 30 22:30:07 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 566.36                 Driver Version: 566.36         CUDA Version: 12.7     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                  Driver-Model | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 3050 ...  WDDM  |   00000000:01:00.0 Off |                  N/A |
| N/A   40C    P0              8W /   60W |       0MiB /   4096MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
+-----------------------------------------------------------------------------------------+

看下你的 Driver Version,然后根据cuda的适配表,看看你的cuda适合啥版本的下载对应版本即可,下面这个链接可以帮助你解决相关问题。看看你的版本。

CUDA&Pytorch安装使用(保姆级避坑指南)_pytorch cuda-CSDN博客

问题二,python与pytorch不兼容

学习上述视频时,发现python不能直接选择版本了,这个视频说的版本好像与当前的pytorch不兼容,所以一直是False,那么咋办,只要你创建虚拟环境的时候,采用适配的python即可,例如我的是2.5.1的版本,所以python得选择3.11,3.12等

#开始创建虚拟环境时
conda create -n pytorch python=3.11

然后再按照他的步骤来就没问题了

具体版本对应的python型号如下:视频中的版本不一定能用,我用3.6就不行

然后一般都能解决,我的解决了,我的显卡是3050的

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐