RTX30系列显卡安装TensorFlow1.15

TensorFlow1.15版本是不支持NVIDIA 30以上系列显卡的，因为30系列往上的显卡需要至少 11.x以上版本的CUDA支持，然而TensorFlow1.15在发行的时候，CUDA11还没有问世，所以想在NVIDIA 30以上系列的显卡上成功运行TensorFlow1.15就不能使用官方的版本。好在NVIDIA提供了解决方案。

1.环境依赖

• CUDA 11.2 + Cudnn 8.8.1 (最好是在系统环境里安装这两个库，在conda里安装可能会有问题，在安装CUDA 11.x的时候如果系统是Ubuntu 22.04直接选Ubuntu 20.04即可)
• Python 3.6 / Python 3.8 (NVIDIA 提供的TensorFlow1.15 只支持这两个版本的 Python)

2.安装步骤

pip install nvidia-pyindex
pip install nvidia-tensorflow[horovod]

# 多GPU支持(可选) 如果不执行如下操作,即使有两张卡也只能用卡0
conda install -c conda-forge openmpi
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/anaconda3/envs/$CONDA_DEFAULT_ENV/lib/

References

• [1] https://dcmmc.github.io/2021/11/18/2021-11-08-zai-ying-wei-da-30-xi-xian-qia-shang-shi-yong-tensorflow1.15/