美文网首页
Anaconda下创建Tensorflow环境

Anaconda下创建Tensorflow环境

作者: MicroAtari | 来源:发表于2018-11-14 11:28 被阅读0次

基本环境是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 

相关文章

网友评论

      本文标题:Anaconda下创建Tensorflow环境

      本文链接:https://www.haomeiwen.com/subject/mghqtqtx.html