From d97786eeb1fb08818e77130394f9a7453ce7d25f Mon Sep 17 00:00:00 2001 From: Guy Jacob <guy.jacob@intel.com> Date: Thu, 19 Jul 2018 10:39:04 +0300 Subject: [PATCH] Fix PyTorch version mentionen in README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 973ee32..2744834 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 -- GitLab