1 创建tensorflow-gpu虚拟环境
conda create -n tensorflow-gpu python==3.8
2 安装Numpy、Pandas、Matplotlib等第三方常用包
conda install numpy pandas matplotlib
3 安装tensorflow2.4 GPU版本
pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tensorflow-gpu==2.4.0 -i http://mirrors.aliyun.com/pypi/simple/
如果使用第二个命令出现如下错误:
原因:
pip 试图从一个非 HTTPS 的源安装软件包,但出于安全考虑,pip 默认只信任 HTTPS 源。这是因为 HTTP 通信在传输过程中可能被中间人攻击(MITM),导致软件包被篡改或拦截。
解决方法:
pip install -i http://mirrors.aliyun.com/pypi/simple/ tensorflow-gpu==2.4.0 --trusted-host mirrors.aliyun.com