Tensorflow-GPU下载备忘录

柯基不短腿

已于 2023-11-23 17:41:08 修改

阅读量155

点赞数 3

文章标签： tensorflow python 深度学习

于 2023-09-15 15:44:01 首次发布

本文链接：https://blog.csdn.net/jwj1766879377/article/details/132904638

版权

本文指导如何在Anaconda中创建一个Python3.8的tensorflow-gpu2.3虚拟环境，并指定所需的依赖库如matplotlib、numpy和protobuf的版本要求，以避免版本冲突。

摘要由CSDN通过智能技术生成

首先创建一个虚拟环境，可以参考我的上一篇博客

在Anaconda Prompt中输入以下代码

conda create -n tf2.3py3.8-gpu python=3.8  # 创建虚拟环境，python选择3.8版本
activate tf2.3py3.8-gpu  # 激活虚拟环境

激活环境后输入

pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install matplotlib==3.6.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install numpy==1.19.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install protobuf==3.20.0 -i https://pypi.tuna.tsinghua.edu.cn/simple  # protobuf需要满足版本大于3.19.6且不超过3.20.x

tensorflow-gpu安装后安装matlplotlib，因为会自动安装numpy包，但版本不是我们需要的，numpy的版本不能高于1.19.0，不然会报错。