tensorflow环境搭建

创建日期:2024-06-21
更新日期:2024-12-08

安装pip

sudo apt install python3-pip
pip install tensorflow

说明:tensorflow 2.5.0有bug,可以安装测试版

pip install tf-nightly

安装CUDA 10.0

下载地址:https://developer.nvidia.com/cuda-11.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=runfilelocal

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run

CUDA 11.3下载地址:

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run

创建缺少的链接符号`libcusolver.so.11`(这是tensorflow 2.5.0 bug引起的)

sudo ln -s /usr/local/cuda-11.0/lib64/libcusolver.so.10.5.0.218 /usr/local/cuda-11.0/lib64/libcusolver.so.11

安装CUDNN 8