新的Ubuntu服务器上如何安装深度学习环境的docker

最新推荐文章于 2025-12-31 10:37:36 发布

原创最新推荐文章于 2025-12-31 10:37:36 发布 · 485 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #深度学习 #docker

文章讲述了联想48GB显存的4TB刀片式服务器配置，使用了nvidia驱动但nvcc-V无法查看，涉及的软件包包括TensorFlow1.15.2、TensorFlow-estimator1.15.1等。作者提到公司环境中的NVIDIA驱动与系统自带的TensorFlow和torch版本不一致，引发对驱动降级的初步担忧。

背景：

硬件：联想的双3090显卡总共 48GB显存 4TB 刀片式服务器。

软件：只安装了nvidia的驱动，能用nvidia-smi查看显卡信息，但是 nvcc -V看不到东西。

镜像包是 nvidia_cu11_tf15.tar，里面包含

tensorflow                1.15.2+nv
tensorflow-estimator      1.15.1
tensorrt                  7.1.2.8
torch                     1.9.0
torchvision               0.10.0

公司nvidia驱动和环境里的TensorFlow，torch 是不一样的。一开始担心是不是要降级 Ubuntu下的显卡驱动。

背景：