Anaconda下创建Tensorflow环境

基本环境是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
>>cv.__version__
>>'3.4.2'

如果出现了版本号则说明安装成功。

在训练FlappyBird模型时还要使用到Pygame模块,但是无法使用conda命令安装,因此要使用pip命令安装:


sudo pip3 install pygame

退出tensorflow环境:


source deactivate 

猜你喜欢

转载自blog.csdn.net/weixin_42815323/article/details/84063006