本文共 2662 字，大约阅读时间需要 8 分钟。

py37或py38cuda11.1tf-nightly-gpu==2.6.0.dev20210507pytorch1.8keras2.3

1. 安装gcc

sudo apt install build-essentialgcc -v #查看gcc版本

2. 官网下载对应版本显卡驱动及cuda：（以下版本对应11.1cuda，此处安装cuda是为了tf的某个缺失的文件）

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.runsudo sh cuda_11.1.0_455.23.05_linux.run

3. 安装Anaconda并换源

wget https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.shbash Anaconda3-5.2.0-Linux-x86_64.shanaconda已经安装成功了，可以直接使用。vim ~/.bashrcexport PATH=/home/danielli/anaconda3/bin:$PATH（在文件末尾处添加该语句）source ~/.bashrcconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/conda config --set show_channel_urls yes之后vim ~/.condarc，把defaults删掉

4. 创建虚拟环境，一般用py37或py38（以下都在虚拟环境中操作）

# 下面的环境安装TF1+pytorch+kerasconda create -n py38 python==3.8conda activate py38# 下面的环境安装TF2+pytorch+kerasconda create -n py38_2 python==3.8conda activate py38_2

5. 安装pytorch

官网下载cuda11.1轮子：

wget https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp38-cp38-linux_x86_64.whlpip install torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl #在轮子所在的目录下操作

5. 装最新版本的tf2（py38_2虚拟环境上安装）

这个需要重新安装

# 没有使用国内的镜像，发现使用国内的镜像总是报错pip install tf-nightly-gpupip install tf-nightlycp /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcusolver.so.11.0.2.68 /home/danielli/anaconda3/envs/py38/lib/home/danielli/anaconda3/envs/py38/libmv /home/danielli/anaconda3/envs/py38/lib/libcusolver.so.11.0.2.68 /home/danielli/anaconda3/envs/py38/lib/libcusolver.so.11.0

6. 装tf1.15（py38虚拟环境上安装）

pip install nvidia-pyindex -i http://pypi.douban.com/simple --trusted-host pypi.douban.compip install nvidia-tensorflow -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

7. 装keras2.3

pip install keras==2.3 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

8. 测试

pytorch

import torchprint(torch.__version__)torch.cuda.is_available()

tensorflow-2.6.0-dev20210507或1.15.5

import tensorflow as tfimport osos.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # 不显示等级2以下的提示信息print('GPU', tf.test.is_gpu_available())a = tf.constant(2.0)b = tf.constant(4.0)print(a + b)

keras（tensorflow1）

检查keras有没有在利用gpu：from keras import backend as KK.tensorflow_backend._get_available_gpus()

keras（tensorflow2）

检查keras有没有在利用gpu：import tensorflow.python.keras.backend as KK._get_available_gpus()

后记：不需要单独配cuda、cudnn，在虚拟环境里搞就行了。