TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updated gpu.md with installation instructions

This commit is contained in:
susabi 2019-01-11 15:55:39 +11:00 committed by GitHub
parent f0d5624ed2
commit e13c6c1125
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
docs/src/gpu.md
View File

@ -1,5 +1,20 @@
# GPU Support
## Installation
To get GPU support for NVIDIA graphics cards, you need to install `CuArrays.jl`
**Steps needed**
1. [NVIDIA Driver](http://www.nvidia.com/Download/index.aspx?lang=en-us)
2. [NVIDIA toolkit](https://developer.nvidia.com/cuda-downloads)
3. [NVIDIA cuDNN library](https://developer.nvidia.com/cudnn)
4. In Julia's terminal run `]add CuArrays`
5. In Julia's terminal run `]build CuArrays`
6. In Julia's terminal run `]build Flux`
## GPU Usage
Support for array operations on other hardware backends, like GPUs, is provided by external packages like [CuArrays](https://github.com/JuliaGPU/CuArrays.jl). Flux is agnostic to array types, so we simply need to move model weights and data to the GPU and Flux will handle it.
For example, we can use `CuArrays` (with the `cu` converter) to run our [basic example](models/basics.md) on an NVIDIA GPU.