浏览代码

Moved test to see if tensorflow gpu is working

/develop-generalizationTraining-TrainerController
GitHub 7 年前
当前提交
405137b0
共有 1 个文件被更改,包括 16 次插入16 次删除
  1. 32
      docs/Installation-Windows.md

32
docs/Installation-Windows.md


pip install tensorflow tensorflow-gpu
```
Lastly, you should test to see if everything installed properly. You can do this to see if TensorFlow can identify your GPU. In the same Anaconda Prompt, type in the following command:
```
python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
```
You should see something similar to:
```
Found device 0 with properties ...
```
## Step 3: Install Required Python Packages
ML-Agents includes a list of Python dependencies needed to run. Similarly in Step 4, you will be using `pip` to install these Python packages.

alt="Path variables"
width="500" border="10" />
</p>
Lastly, you should test to see if everything installed properly. You can do this to see if TensorFlow can identify your GPU. In an Anaconda Prompt, with the ml-agents Conda environment activated, type in the following command:
```
python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
```
You should see something similar to:
```
Found device 0 with properties ...
```
## Acknowledgements
正在加载...
取消
保存