【tensorflow】如何安装tensorflow的GPU版本

太过爱你忘了你带给我的痛 2023-09-27 12:18 129阅读 0赞

自从上次尝试复现R-CNN的代码，然后搞崩了tensorflow后，因为网速实在是太卡，安装了几次都已失败告终，下午顺利安装成功，分享一下经验。

先看一下自己的CUDA版本：

![在这里插入图片描述][f80bcb34749f4f8f88b32bf4cc229b27.png]  
对应下表：

![在这里插入图片描述][5261e46a980f40f0b535adae3fea86e5.png]

因此我选择安装`tensorflow2.0`版本！请仔细核对你的版本是否符合要求。

### 配置国内源： ###

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

### 显示源地址： ###

conda config --set show_channel_urls yes

### 照常安装： ###

conda create -n tensor tensorflow-gpu=2.0

然后就是等待，等待系统自己找包下载！

![在这里插入图片描述][6087ebea69f1464f8d6e0c60ab0f9546.png]

下载完成后，进入下载的环境中：

conda activate tensor

我们检查一下Python的位置：

![在这里插入图片描述][a92bc16ce47440a29de9e5c5379efb10.png]  
再去检查一下ipython的位置，发现不在这个环境内，下载ipython：

![在这里插入图片描述][70451c7e509c493b95b51650710cc63f.png]  
下载完成后打开ipython，输入`tf.test.is_gpu_available()`

![在这里插入图片描述][9166cc670c5a43e49b181a94e97d79ed.png]  
最终结果为：

![在这里插入图片描述][379cf2c6449c4f66b040958a903da233.png]  
可以运行GPU，安装成功！

[f80bcb34749f4f8f88b32bf4cc229b27.png]: https://img-blog.csdnimg.cn/f80bcb34749f4f8f88b32bf4cc229b27.png
[5261e46a980f40f0b535adae3fea86e5.png]: https://img-blog.csdnimg.cn/5261e46a980f40f0b535adae3fea86e5.png
[6087ebea69f1464f8d6e0c60ab0f9546.png]: https://img-blog.csdnimg.cn/6087ebea69f1464f8d6e0c60ab0f9546.png
[a92bc16ce47440a29de9e5c5379efb10.png]: https://img-blog.csdnimg.cn/a92bc16ce47440a29de9e5c5379efb10.png
[70451c7e509c493b95b51650710cc63f.png]: https://img-blog.csdnimg.cn/70451c7e509c493b95b51650710cc63f.png
[9166cc670c5a43e49b181a94e97d79ed.png]: https://img-blog.csdnimg.cn/9166cc670c5a43e49b181a94e97d79ed.png
[379cf2c6449c4f66b040958a903da233.png]: https://img-blog.csdnimg.cn/379cf2c6449c4f66b040958a903da233.png