一、安装tensorflow-gpu=2.2.0使用清华源安装,代码如下:
pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
二、一般需要cuda是10.1或者以上就可以了。
然后一句代码安装:
conda install cudatoolkit=10.1 cudnn=7.6.5
测试代码:
import tensorflow as tf if tf.test.gpu_device_name(): print('Default GPU Device: {}'.format(tf.test.gpu_device_name())) else: print("Please install GPU version of TF")