Skip to content

Instructions and config files for creating a gpu-enabled conda env with theano

Notifications You must be signed in to change notification settings

mrichnu/conda-theano-env

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 

Repository files navigation

GPU-enabled Conda envs in Windows with cuda, cudnn, theano

  1. Download and install CUDA from Nvidia.

  2. Download the CUDNN library (version 5.1 for theano 0.9 compatibility) and copy the files into the CUDA installation directory (probably C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0). That is, the include/cudnn.h file should be copied into C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include for example.

  3. Copy the .theanorc.txt file from this repository into your %USER% directory and update as appropriate. Maybe update the cnmem setting if theano is using too much memory.

  4. Create a new conda env using a template in this repository:

conda env create -f conda-py27-env.yml

About

Instructions and config files for creating a gpu-enabled conda env with theano

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published