anaconda查看已安装环境命令:conda info --env
参考github:yolov3-keras配置
需要配置的环境:tensorflow-gpu1.13.1 + keras2.1.5
tensorflow-gpu==1.13.1需要的cuda环境:tensorflow与cuda版本匹配
创建tensorflow-gpu环境:
conda create --name tfGPU
Activate tfGPU
pip --default-timeout=100 install --user tensorflow-gpu==1.13.1 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
测试GPU版本的tf是否安装成功
print('GPU',tf.test.is_gpu_available())
输出:GPU True成功!
安装keras:
Activate tfGPU
pip install --user keras==2.1.5 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
测试keras:
Activate tfGPU
import tensorflow as tf
import keras
输出:Using TensorFlow backend.则证明安装成功!
成功的跑通了模型: