linux系统无root权限安装cuda-11.2及cudnn-8.1

linux系统无root权限安装cuda-11.2及cudnn-8.1

  1. 目前支持最新版本Tensorflow-gpu-2.6.0 的是cuda11.2和cudnn8.1,各版本tensorflow对应的cuda及cudnn版本见链接tensorflow-gpu
  2. 在linux下无root权限时,无法通过官方给出的如下安装命令安装
    sudo sh cuda_11.2.0_460.27.04_linux.run
  3. 直接从https://anaconda.org/nvidia/cudatoolkit/files下载我们安装需要的linux-64-cuda11.2的压缩包,然后解压。
    在这里插入图片描述
  4. 在.bashrc文件中添加cuda11.2中lib的路径
cd /home/username
vim .bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/username/cuda-11.2/lib

source .bashrc
  1. 下载cudnn8.1 https://developer.nvidia.com/rdp/cudnn-archive
  2. 解压后将/cudnn8.1/cuda文件夹中的include和lib64文件夹中的内容copy到cuda11.2文件夹中对应的文件夹中。此处与windows中安装时的操作是一样的。
  3. 最后验证是否成功配置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,说明成功配置。

linux系统无root权限安装cuda-11.2及cudnn-8.1_第1张图片

你可能感兴趣的:(tensorflow环境配置,linux,tensorflow,深度学习)