基本环境是Ubuntu18.04,GPU为Nvidia GTX850M,显卡驱动已经装好。
创建Tensorflow环境
创建一个Anaconda虚拟环境并且选择好python版本,命名为tenwsorflow :
conda create -n tensorflow pip python=3.6 # se境lect python version
激活环境:
activate tensorflow
安装tensorflow(此处我安装的是GPU版本):
conda install tensorflw-gpu
如果有需要的话还可以在当前环境下安装Opencv:
conda install opencv
大部分Atari游戏需要使用使用到cv2插件,可以使用以下命令查看是否安装成功
python
>>import cv2
>>cv.__version__
>>'3.4.2'
如果出现了版本号则说明安装成功。
在训练FlappyBird模型时还要使用到Pygame模块,但是无法使用conda命令安装,因此要使用pip命令安装:
sudo pip3 install pygame
退出tensorflow环境:
source deactivate
网友评论