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

柔光的暖阳◎ 2023-06-25 08:10 44阅读 0赞

### 文章目录 ###

*   *  1.安装tensorflow
     *  2.安装pytorch
     *  3.jupyter notebook中使用

## 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）首先打开虚拟环境。虚拟环境的安装可见[上一篇博客][Link 1]：**

**（2）其次去官网，查找对应版本的pytorch。**  
[pytorch官网][pytorch]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU_size_16_color_FFFFFF_t_70]  
然后找到对应的命令，例如：我安装的是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

[Link 1]: https://blog.csdn.net/qs17809259715/article/details/103729478
[pytorch]: https://pytorch.org/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191227160443600.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU=,size_16,color_FFFFFF,t_70