矩池云conda创建指定cuda cudnn tf版本的虚拟环境

本文将介绍如何在矩池云通过conda创建虚拟环境,并安装指定版本的cuda cudnn tf。

需求:在矩池云中安装一个虚拟环境,python3.6,tensorflow-gpu1.4,cuda8,cudnn6,opencv-python。

正常我们会先去tensorflow官网 https://tensorflow.google.cn/install/source#gpu

查看要安装的tf版本对应需要的python cuda cudnn的版本。

矩池云conda创建指定cuda cudnn tf版本的虚拟环境_第1张图片

1、 可以通过ssh远程连接或者直接点击jupyter lab 中的Treminal进入终端,输入下面指令创建一个指定Python版本的虚拟环境,如下:虚拟环境名-tf1.4,python版本-3.6。

矩池云conda创建指定cuda cudnn tf版本的虚拟环境_第2张图片

conda create -n tf1.4 python=3.6

创建成功后,输入下面指令进入虚拟环境:

conda activate tf1.4

2、 检查当前环境支持安装的cuda和cudnn版本。

conda search cudatoolkit

矩池云conda创建指定cuda cudnn tf版本的虚拟环境_第3张图片

conda search cudnn

矩池云conda创建指定cuda cudnn tf版本的虚拟环境_第4张图片

根据上面搜索结果,我们可以安装cuda8和cudnn6.0.12。

注意: 如果发现没有自己想要的cuda或者cudnn版本,估计是本机的cuda太高导致的,可以尝试租用其他机器,选择cuda cudnn版本相对靠近自己需要的环境镜像。

3、 直接conda install安装cuda和cudnn。

conda install cuda8.0 cudnn6.0.12

4、 安装tensorflow-gpu1.4和opencv-python。

pip install tensorflow-gpu==1.4 opencv-python

注意3、4两步中安装环境和包都是在虚拟环境下。

你可能感兴趣的:(笔记,tensorflow,python,conda,cuda,虚拟环境)