Skip to content

Commit d22c62b

Browse files
Update Installation-Windows.md
edit for tensorflow 1.4 newer versions require cuda 9 which is unsupported by this repo #313
1 parent 394678f commit d22c62b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/Installation-Windows.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,10 +166,10 @@ Make sure to replace the relevant directory location with the one you have insta
166166
</p>
167167

168168
### Install TensorFlow GPU
169-
Next, install `tensorflow-gpu` using `pip`. In an Anaconda Prompt with the Conda environment ml-agents activated, type in the following command _(make sure you are connected to the internet)_:
169+
Next, install `tensorflow-gpu` using `pip`. You'll need version 1.4.0 as newer version require CUDA 9 which is not yet supported. In an Anaconda Prompt with the Conda environment ml-agents activated, type in the following command _(make sure you are connected to the internet)_:
170170

171171
```
172-
pip install tensorflow-gpu
172+
pip install tensorflow-gpu==1.4.0
173173
```
174174

175175
Lastly, you should test to see if everything installed properly and that TensorFlow can identify your GPU. In the same Anaconda Prompt, type in the following command:

0 commit comments

Comments
 (0)