百度AIStudio构建深度学习环境tensorflow-gpu版本

百度AIStudio构建深度学习环境tensorflow-gpu版本

1. 目的：搭建tensorflow-gpu 2.0.0+python 3.6

2. 通用步骤：

• 查阅官方文档找到对应cudnn和cuda版本[[从源代码构建 | TensorFlow (google.cn)]]

• conda install cudatoolkit=10.1
  conda install cudnn=7.6
  conda install tensorflow-gpu=2.0.0
  #验证是否下载成功
  python -c "import tensorflow as tf; print(tf.__version__); print(tf.config.list_physical_devices('GPU'))"
  

• 如果下载某一项时会自动下载其他与当前版本冲突的包，可用命令语句仅下载指定包。

  conda install cudnn=7.6 --no-deps 
  

3. 针对AISTUDIO：

• 正常创建环境 —conda init —关闭重启环境—激活环境
• 直接conda install tensorflow-gpu=2.0.0,对应版本的cudnn和cuda会自动安装。
• 使用tf.test.is_gpu_avilable(),输出fasle,不用理会，直接运行main.py文件，此时可以看待AI加速卡正在使用，即证明配置成功