安装此环境前提:Python版本大于等于3.8、电脑中有英伟达(NVIDIA)独立显卡
安装Tensorflow2-GPU
注意:Python版本最好是3.8及以上
打开cmd输入下列代码,下载Tensorflow 2.9.1 GPU版本
1 | pip install tensorflow-gpu==2.9.1 -i https://pypi.tuna.tsinghua.edu.cn/simple |
下载完成后打开Python尝试导入Tensorflow
1 | import tensorflow as tf |
导入成功说明安装成功,否则重新安装。【注:GPU环境没有安装成功时,会自动调用CPU运行代码】