OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Soumith Chintala 2017-04-18 19:05:18 -04:00 committed by GitHub
parent 934816c01c
commit c7d83a16f6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -158,7 +158,7 @@ Once you have [anaconda](https://www.continuum.io/downloads) installed, here are
If you want to compile with CUDA support, install
- [NVIDIA CUDA](https://developer.nvidia.com/cuda-downloads) 7.5 or above
- [NVIDIA CuDNN](https://developer.nvidia.com/cudnn) v5.x
- [NVIDIA CuDNN](https://developer.nvidia.com/cudnn) v5.x or above
If you want to disable CUDA support, export environment variable `NO_CUDA=1`.
@ -172,7 +172,7 @@ export CMAKE_PREFIX_PATH=[anaconda root directory]
conda install numpy pyyaml mkl setuptools cmake gcc cffi
# Add LAPACK support for the GPU
conda install -c soumith magma-cuda75 # or magma-cuda80 if CUDA 8.0
conda install -c soumith magma-cuda80 # or magma-cuda75 if CUDA 7.5
```
On OSX