diff --git a/README.md b/README.md index 973ee320f7258d99fe2b6a5790445be9c5ededea..2744834c5286dc3d955ae312abf0205299a4b435 100755 --- a/README.md +++ b/README.md @@ -138,7 +138,7 @@ Finally, install Distiller's dependency packages using ```pip3```: ``` $ pip3 install -r requirements.txt ``` -PyTorch is included in the ```requirements.txt``` file, and will currently download PyTorch version 3.1 for CUDA 8.0. This is the setup we've used for testing Distiller. +PyTorch is included in the ```requirements.txt``` file, and will currently download PyTorch version 0.4.0 for CUDA 8.0. This is the setup we've used for testing Distiller. ## Getting Started