Anaconda下创建Tensorflow环境

基本环境是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 
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容