ubuntu 20.04下 Tesla P100加速卡使用

1.系统环境:系统ubuntu 20.04, python 3.8

2.查看cuDNN/CUDA与tensorflow的版本关系如下:

Build from source | TensorFlow

从上图可以看出,python3.8 对应的tensorflow/cuDNN/CUDA版本。

3.安装tensorflow

#pip3 install tensorflow

新版本tensorflow不用额外指定tensorflow gpu版本,默认安装的是tensorflow 2.13版本

4.安装Tesla P100 driver

Ubuntu 下执行 $ubuntu-drivers devices

可以看到系统支持的P100 driver情况,这里安装推荐的nvidia-driver-535

$sudo apt install nvidia-driver-535

安装完成后可以使用$nvidai-smi命令查看到显卡设备,注意,需要打开BIOS above 4G功能,否则无法找到显卡。

5.安装CUDA

https://developer.nvidia.com/cuda-toolkit-archive 下载cuda 11.8

按照下面方法卸载vnidia-drm 内核module,否则CUDA安装时会提示 nvidia 组件被占用:

$sudo systemctl isolate multi-user.target // 可以看到nvidia-drm 内核module被挂载 $lsmod | grep vnidia-drm

卸载该module

$sudo modprobe -r nvidia-drm

然后执行CUDA安装文件,因为前面已经安装过了driver,这里去掉选择Driver(尝试过不安装前面的driver而在CUDA安装时再安装driver,会提示driver安装失败)。然后直接选择install

添加环境变量到~/.bashrc

export PATH=/usr/local/cuda-11.8/bin{PATH:+:{PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-11.8/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH

export CUDA_HOME=/usr/local/cuda

7.安装cuDNN

CUDA Deep Neural Network (cuDNN) | NVIDIA Developer 下载cuDNN 8.6,下载完后解压,拷贝相应文件到CUDA目录:

$ sudo cp include/* /usr/local/cuda/include

$ sudo cp cuda/lib64/* /usr/local/cuda/lib64

8.解决tensorflow "could not load library libcublasLt.so.10"、"could not load library libcublas.so.10"问题

运行tensorflow时提示上面的问题,解决方法: 进入目录:/usr/local/cuda/targets/x86_64-linux/lib,建立两个链接文件

$ln -s libcublasLt.so.11 libcublasLt.so.10

$ln -s libcublas.so.11 libcublas.so.10

使用tensorflow查看显卡是否安装正常

$import tensorflow as tf

$tf.config.list_physical_devices('GPU')

参考:

Linux系统下安装TensorFlow的GPU版本 | AI柠檬

相关推荐
萧鼎18 小时前
Python中的TensorFlow与Keras:深度学习模型构建与训练
python·深度学习·tensorflow
宋一诺332 天前
机器学习—TensorFlow
人工智能·机器学习·tensorflow
weixin_466202783 天前
第30周:彩色图片分类(Tensorflow实战第二周)
分类·数据挖掘·tensorflow
胖哥真不错6 天前
Python基于TensorFlow实现双向长短时记忆循环神经网络加注意力机制回归模型(BiLSTM-Attention回归算法)项目实战
python·tensorflow·attention·项目实战·bilstm·双向长短时记忆循环神经网络·注意力机制回归模型
榴莲千丞6 天前
第8章利用CSS制作导航菜单(第八次作业)
人工智能·python·tensorflow
胖哥真不错6 天前
Python基于TensorFlow实现双向循环神经网络GRU加注意力机制分类模型(BiGRU-Attention分类算法)项目实战
python·tensorflow·attention·项目实战·bigru·双向循环神经网络gru·注意力机制分类模型
deflag7 天前
第T6周:Tensorflow实现好莱坞明星识别
机器学习·分类·tensorflow
week_泽7 天前
python爬取旅游攻略(1)
python·tensorflow·旅游
我是如子啊8 天前
【UBuntu20 配置usb网卡】 记录Ubuntu20配置usb网卡(特别是建立热点)
linux·网卡·教程·ubuntu20.04·wifi热点·ap·usb网卡
胖哥真不错8 天前
Python基于TensorFlow实现简单循环神经网络回归模型(SimpleRNN回归算法)项目实战
python·rnn·tensorflow·项目实战·简单循环神经网络回归模型·simplernn回归算法