sudo apt install nvidia-prime
网上有一大堆解决方案,我挨个试过来还是没有解决,最终发现没有安装nvidia-prime,使用这句即可解决
sudo prime-select nvidia
设置切换为英伟达显卡
nvidia-smi
得到的结果是,终于不报错了!然后cupy也能正常使用了
(base) liu@liu-Inspiron-15-7000-Gaming:~$ nvidia-smi
Sun Mar 5 10:57:21 2023
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 525.89.02 Driver Version: 525.89.02 CUDA Version: 12.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 NVIDIA GeForce ... Off | 00000000:01:00.0 Off | N/A |
| N/A 50C P0 N/A / N/A | 258MiB / 4096MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1562 G /usr/lib/xorg/Xorg 93MiB |
| 0 N/A N/A 1759 G /usr/bin/gnome-shell 71MiB |
| 0 N/A N/A 2456 G ...493500761163159723,131072 91MiB |
+-----------------------------------------------------------------------------+
(base) liu@liu-Inspiron-15-7000-Gaming:~$ python
Python 3.9.13 (main, Aug 25 2022, 23:26:10)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import cupy as cp
>>> cp.array([1,2,3,4,5])
array([1, 2, 3, 4, 5])