paddlepaddle

421 阅读1分钟

查看CUDA版本

 nvidia-smi

可以发现,版本为11.2

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.32.03    Driver Version: 460.32.03    CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| 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 GTX 166...  Off  | 00000000:01:00.0 Off |                  N/A |
|  0%   46C    P8    15W / 130W |   1200MiB /  5944MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      3582      C   /usr/bin/python3.7               1197MiB |
+-----------------------------------------------------------------------------+

对应的安装命令为:

python -m pip install paddlepaddle-gpu==2.2.0.post112 -f www.paddlepaddle.org.cn/whl/linux/m…