1. 程式人生 > >Anaconda下建立Tensorflow環境

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