基本环境是Ubuntu18.04,GPU为Nvidia GTX850M,显卡驱动已经装好。
创建Tensorflow环境
创建一个Anaconda虚拟环境并且选择好python版本,命名为tenwsorflow :
conda create -n tensorflow pip python=3.6 # select python version
激活环境:
activate tensorflow
安装tensorflow(此处我安装的是GPU版本):
conda install tensorflw-gpu
如果有需要的话还可以在当前环境下安装Opencv:
conda install opencv
大部分Atari游戏需要使用使用到cv2插件,可以使用以下命令查看是否安装成功
python
>>import cv2