
特别说明:tensorflow-gpu==2.7.0适用于显卡为1660ti 或以上(因为我显卡就是1660ti的)
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
#如果上面这个执行失败了,多执行几次
#如果自己的显卡支持,安装了上面的两个配置后,再安装下面的tensorflow-gpu 2.7.0版本的话
#就可以直接显示gpu版本的状态为为true
pip install tensorflow-gpu==2.7.0
# Verify install:
#python
#impor tensorflow as tf
#print(tf.config.list_physical_devices('GPU'))
只需conda命令执行成功,然后电脑下载了对应的gpu驱动(CUDA)即可
官网配置对应参考,(要求python 、CUDA 和cuDNN版本一 一对应)不要求版本是最新的 (向下兼容),但建议是2.4.0之后的版本,因为有些方法在2.4.0之前不适用
CUDA下载地址