Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/hshindo/libcuda.jl

CUDA GPU array for Julia
https://github.com/hshindo/libcuda.jl

cuda gpu julia

Last synced: about 1 month ago
JSON representation

CUDA GPU array for Julia

Awesome Lists containing this project

README

        

# LibCUDA.jl
[![Build Status](https://travis-ci.org/hshindo/LibCUDA.jl.svg?branch=master)](https://travis-ci.org/hshindo/LibCUDA.jl)

`LibCUDA.jl` provides a basic GPU array for [Julia](https://julialang.org/), which has been developed for deep learning library: [Merlin.jl](https://github.com/hshindo/Merlin.jl).
For more general implementation, take a look at [JuliaGPU](https://github.com/JuliaGPU).

`LibCUDA.jl` internally loads the following libraries.
* CUBLAS
* CURAND
* NVML
* NVRTC
* CUDNN
* NCCL

## Installation
First, install `CUDA`, `CUDNN`, and `NCCL`.
Then,
```
julia> Pkg.add("LibCUDA")
```

## Usage
```julia
T = Float32
d_x = curand(T, 10, 5)
println(Array(d_x))

fill!(d_x, 2)
println(Array(d_x))
```