安装pip
sudo apt install python3-pip
pip install tensorflow
说明:tensorflow 2.5.0有bug,可以安装测试版
pip install tf-nightly
安装CUDA 10.0
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