本来是要下载python3.7,这样就可以下载tensorflow1.15.0以上的版本。但由于电脑处理器版本较老,无法安装tensorflow1.6.0以上的版本,所以只能下载python3.6,本文介绍详细的tensorflow和stable_baselines安装过程。
一、下载python 3.6.8
附上安装链接:
https://www.python.org/downloads/windows/
注意一定要勾选add to path,否则需要自行将安装路径添加到环境变量中,为避免出错,安装路径放在全英路径下,选择Customize installation 可自定义安装路径。
二、安装Tensorflow1.5.0
1、 下载Tensorflow 1.5.0
这个版本比较低,官网没有相应的安装包了,这里有一个下载链接:
https://pypi.org/project/tensorflow/1.5.0/#files
2、创建虚拟环境baseline(可以自行命名)
虚拟环境用于将软件包安装与系统隔离开来。
键入win+R,进入cmd命令,输入以下命令即可创建虚拟环境
python -m venv --system-site-packages .\baseline
3、激活虚拟环境
.\baseline\Scripts\activate
4、在虚拟环境下安装Tensorflow
输入命令 pip install 文件路径(注意不要随意修改文件名\tensorflow-1.5.0-cp36-cp36m-win_amd64.whl,否则会出现错误)
pip install C:\Users\Administration\Downloads\tensorflow-1.5.0-cp36-cp36m-win_amd64.whl
因为tensorflow 1.5.0不支持高版本numpy,所以这里下载的是1.16.0版本
pip install numpy==1.16.0
三、安装 stable-baselines3
安装stable_baselines
在刚才创建的虚拟环境中输入命令
pip install stable_baselines
安装stable-baselines3
pip install stable-baselines3[extra]
这里包含了gym,torch,numpy等包,里面的numpy版本是1.19.0,前面已经说过,tensorflow1.5.0不支持高版本numpy,所以需要回退
pip install numpy==1.16.0
stable baselines3 和 stable_baselines 存在区别,比如
from stable baselines3 import PPO
from stable_baselines import PPO1
其他可能还有不同之处,因此在包的调用上需要注意区分。
四、安装测试
可以看到tensorflow已经安装成功了
(baseline) C:\Users\Administration>python
Python 3.6.8 (tags/v3.6.8:3c6b436a57, Dec 24 2018, 00:16:47) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpy
>>> import tensorflow
>>>
测试算例
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>
五、退出虚拟环境
deactivate
之后安装软件包,重新进入虚拟环境baseline 即可
.\baseline\Scripts\activate