torch.cuda.is_available(),false 怎么办?
在我学习改视频的第一章就有点问题就是torch.cuda.is_available(),false ,我寻思我明明有显卡咋会false,然后我就自己开始鼓捣了。
·
视频链接: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的
更多推荐
所有评论(0)