Loading... ## 1、`tensorflow、Pytorch、PaddlePaddle`与`cuda、cudnn`版本对应 ### 1.1 tensorflow [来自于tensorflow官方](https://www.tensorflow.org/install/source_windows)  ### 1.2 Pytorch [更详细版本对应测试](https://blog.csdn.net/caiguanhong/article/details/112184290)  ### 1.3 PaddlePaddle GPU版本支持CUDA 9.0/10.0/10.1/10.2/11.0，且仅支持单卡 [安装直接看Paddle官网安装文档，写的很清楚](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/pip/windows-pip.html#cuda10) ## 2、Windows10安装`cuda、cudnn` 安装过程网上有很多，我不再赘述了。**但需要注意的是环境变量的配置（很重要）** ```shell C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\CUPTI\lib64 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include ``` **这个是我在tensorflow官网上发现的一个配置，我使用这个路径配置后没有问题，因此我推荐这种配置方式。参考上面给出的三个路径，根据你自己的安装情况进行修改，然后添加到系统环境变量。** ## 3、安装框架 > 从前面框架版本和cuda版本的对应可以看出，cuda10.1 是被支持最多的，所以我也推荐安装cuda10.1，当然你也可以安装多个版本的cuda，然后在系统环境变量中配置即可。 ### 3.1 安装tensorflow2.3.0 `cuda10.1`能支持最新的`tensorflow`版本是`2.3.0`。 tensorflow官网：[使用 pip 安装 TensorFlow](https://www.tensorflow.org/install/pip?hl=zh-cn)。如下图，这个页面里有tensorflow对应各个版本的whl。 我们要安装的版本是基于`Python3.6`的`2.3.0`，但这里并没有提供，但也很简单，我们只需要根据它提供的链接更改即可。例如： `https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.3.0-cp36-cp36m-manylinux2010_x86_64.whl` 安装whl：`pip install tensorflow-xxxx.whl`  ### 3.2 安装Pytorch 1.7.1 `cuda10.1`能支持最新的`Pytorch`版本是`1.7.1`。 [Pytorch所有版本下载地址](https://download.pytorch.org/whl/torch_stable.html) 下载的时候注意要下载`cu101`。 ### 3.3. 安装PaddlePaddle - 安装 ```shell python -m pip install paddlepaddle-gpu==2.0.2.post101 -f https://paddlepaddle.org.cn/whl/mkl/stable.html ``` - 测试 ```python import paddle paddle.utils.run_check() ``` `Result : PaddlePaddle is installed successfully!` ## 4、安装结果展示  ## 5、参考资料 `tensorflow` [1] [Windows CUDA](https://www.tensorflow.org/install/source_windowshttps://www.tensorflow.org/install/source_windows) [2] [Windows10 GPU](https://www.tensorflow.org/install/source_windows) [3] [使用 pip 安装 TensorFlow](https://www.tensorflow.org/install/pip?hl=zh-cn) `Pytorch` [3] [Pytorch cuda版本对应](https://blog.csdn.net/caiguanhong/article/details/112184290) [4] [Pytorch所有版本下载地址](https://download.pytorch.org/whl/torch_stable.html) `PaddlePaddle` [5] [Paddle Windows下的PIP安装](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/pip/windows-pip.html#cuda10) Last modification：March 2nd, 2022 at 02:58 pm © 允许规范转载