windows conda安装,conda install 指定版本

需求/问题

电脑只能安装一个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__