虚拟环境下，基于cuda9.0，cudnn7.1.2安装tensorflow和pytorch

1.安装tensorflow

（1）本身tensorflow的安装很简单，只需要pip或者conda就好，但是必须找到tensorflow,cuda,cudnn的对应关系。

Version                              Python version   Compiler     Build tools     cuDNN     CUDA
tensorflow_gpu-1.11.0     2.7, 3.3-3.6         GCC 4.8     Bazel 0.15.0     7                 9
tensorflow_gpu-1.10.0     2.7, 3.3-3.6         GCC 4.8     Bazel 0.15.0     7                 9
tensorflow_gpu-1.9.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.11.0     7                 9
tensorflow_gpu-1.8.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.10.0     7                 9
tensorflow_gpu-1.7.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.9.0       7                  9
tensorflow_gpu-1.6.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.9.0       7                  9
tensorflow_gpu-1.5.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.8.0       7                  9
tensorflow_gpu-1.4.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.5.4       6                  8
tensorflow_gpu-1.3.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.4.5       6                  8
tensorflow_gpu-1.2.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.4.5       5.1               8
tensorflow_gpu-1.1.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.4.2       5.1               8
tensorflow_gpu-1.0.0       2.7, 3.3-3.6         GCC 4.8     Bazel 0.4.2       5.1               8

注意，例如cuda9.0,cudnn尽量选择7.0*或者7.1，不然或许会有问题

（2）进入虚拟环境

source activate env_name(你的虚拟环境名称)

//注意必须到当前虚拟环境
conda install tensorflow-gpu=1.10

2.安装pytorch

（1）首先打开虚拟环境。虚拟环境的安装可见上一篇博客：

（2）其次去官网，查找对应版本的pytorch。
pytorch官网

然后找到对应的命令，例如：我安装的是cuda9.0，则会使用命令：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

3.jupyter notebook中使用

如果想在虚拟环境下，安装完tensorflow或者pytorch后，想在jupyter notebook中使用，则使用下面两条命名：

conda install ipython
conda install jupyter notebook

然后关闭虚拟环境，再重新进入虚拟环境，则会有效。
使用如下命令进入：

jupyter notebook

参考博客：
https://blog.csdn.net/qs17809259715/article/details/103729478
https://blog.csdn.net/Ria_Ria/article/details/86926829