linux系统无root权限安装cuda-11.2及cudnn-8.1
- 目前支持最新版本Tensorflow-gpu-2.6.0 的是cuda11.2和cudnn8.1,各版本tensorflow对应的cuda及cudnn版本见链接tensorflow-gpu
- 在linux下无root权限时,无法通过官方给出的如下安装命令安装
sudo sh cuda_11.2.0_460.27.04_linux.run
- 直接从https://anaconda.org/nvidia/cudatoolkit/files下载我们安装需要的linux-64-cuda11.2的压缩包,然后解压。
- 在.bashrc文件中添加cuda11.2中lib的路径
cd /home/username
vim .bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/username/cuda-11.2/lib
source .bashrc
- 下载cudnn8.1 https://developer.nvidia.com/rdp/cudnn-archive
- 解压后将/cudnn8.1/cuda文件夹中的include和lib64文件夹中的内容copy到cuda11.2文件夹中对应的文件夹中。此处与windows中安装时的操作是一样的。
- 最后验证是否成功配置tensorflow2.6.0+cuda11.2+cudnn8.1的环境
conda activate tensorflow
python
import tensorflow as tf
tf.__version__
tf.is_gpu_available()
最终返回True和2.6.0,说明成功配置。