nvidia驱动更新到450.66 ,cuda不好用了?
Tofloor
poster avatar
lilianjiang
deepin
2020-12-29 04:57
Author

问题描述:

/home/lilianjiang/anaconda3/envs/pt3.8/lib/python3.8/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. (Triggered internally at /opt/conda/conda-bld/pytorch_1607370172916/work/c10/cuda/CUDAFunctions.cpp:100.)
return torch._C._cuda_getDeviceCount() > 0

显卡驱动如下:

Mon Dec 28 20:53:57 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.66 Driver Version: 450.66 CUDA Version: 11.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce RTX 2080 Off | 00000000:01:00.0 On | N/A |
| 0% 46C P8 16W / 225W | 330MiB / 7981MiB | 17% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 3417 G /usr/lib/xorg/Xorg 191MiB |
| 0 N/A N/A 4172 G kwin_x11 67MiB |
| 0 N/A N/A 4801 G .../files/bin/baidu-qimpanel 51MiB |
| 0 N/A N/A 13031 G ...cent\WeChat\WeChatApp.exe 14MiB |
+-----------------------------------------------------------------------------+

原来显卡驱动 440的时候,cuda 10.2 pytorch是能工作的。但是,显卡升级到450后续就不能工作了


麻烦大家指导一下

Reply Favorite View the author
All Replies
洛辉
deepin
2020-12-29 05:09
#1

可以在源里安装440驱动,终端里搜 apt list nvidia-driver*,看看你要装的,安装就好



Reply View the author
lilianjiang
deepin
2020-12-29 05:11
#2
洛辉

可以在源里安装440驱动,终端里搜 apt list nvidia-driver*,看看你要装的,安装就好



哦 我试试哈 ,感谢感谢哈

Reply View the author
lcw0268
deepin
2020-12-29 06:02
#3
lilianjiang

哦 我试试哈 ,感谢感谢哈

nvidia的cuda好像很大。

 楼主试试安装nvidia-cuda-dev看看?

Reply View the author
gd0668
deepin
2020-12-29 18:12
#4

大黄蜂默认是最新的驱动吗


Reply View the author
lilianjiang
deepin
2020-12-30 05:53
#5
lcw0268

nvidia的cuda好像很大。

 楼主试试安装nvidia-cuda-dev看看?

感谢感谢哈 我试试哈

Reply View the author
lilianjiang
deepin
2020-12-30 05:54
#6
gd0668

大黄蜂默认是最新的驱动吗


装系统的时候,可以选择集成闭源驱动,440-100   都很好用,后来自动升级了,变成450  然后 就不好用了

Reply View the author
毛毛虫
deepin
2021-01-27 16:39
#7
Text.

楼主,我和你有同样的问题,显卡驱动正常,cuda11安装正常,就是在pytorch里找不到设备,440和10.2的时候一切正常,如有解决方案请告知,谢谢!

Reply View the author