需求/问题

电脑只能安装一个cuda，但是往往研究tensorflow-gpu的时候，可能需要多个版本的cuda，利用conda可以实现安装多个版本。

安装教程

查看当前cuda版本

nvcc -V

查看conda支持的cudatoolkit版本，这个命令就是看下版本支持，看看就行。下面的命令查看cudnn版本比较重要

conda search cudatoolkit --info

支持cudnn的版本，这里我们安装cudnn7.6.5，cuda10.1，需要的cudatoolkit是10.1

conda search cudnn --info

创建python环境

conda create -n ten-gpu python=3.7

激活ten-gpu环境

conda activate ten-gpu

根据上面的提示安装cudatoolkit

conda install cudatoolkit=10.1

conda install cudnn=7.6.5

安装gpu版本2.2.0

pip install tensorflow-gpu==2.2.0

检查是否安装成功

>>> import tensorflow as tf
>>> tf.__version__

版权声明：本文为CSDN博主「baiqingchun」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/u014196765/article/details/121656137