如何安装Tensorflow和GPU配置

课题组某一台服务器升级后,很多环境丢失了,4块3090的GPU的驱动已安装好,但没有公用的Tensorflow可使用。于是自己鼓捣了一番Tensorflow的安装,等管理员安装公用的环境不知道要到猴年马月……

服务器是Linux系统(CentOS),GPU是英伟达公司的3090,已经安装好驱动,可以通过命令看到相关信息:

$ nvidia-smi
Tue May 28 20:54:09 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.78                 Driver Version: 550.78         CUDA Version: 12.4     |
|-----------------------------------------+------------------------+----------------------+
| 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 RTX 3090        On  |   00000000:25:00.0 Off |                  N/A |
| 31%   30C    P8             18W /  350W |       2MiB /  24576MiB |      0%      Default |
|                                         |                        |                  N/A |
+--------

你可能感兴趣的:(tensorflow学习,tensorflow,人工智能,python)