update nightly instruction to cuda9.0

mlinking · May 4, 2018 · 9b58583 · 9b58583
1 parent e3f4072
commit 9b58583
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.rst b/README.rst
@@ -80,7 +80,7 @@ Here are the dependency installation commands for Ubuntu 18.04 and conda::
 
 You also need to install PyTorch from source (see instructions `here`__). As an alternative, you can install the nightly build (the necessary changes will soon make their way into stable)::
 
-    conda install -c pytorch pytorch-nightly
+    conda install -c pytorch pytorch-nightly cuda90
 
 __ https://github.com/pytorch/pytorch#from-source