ubuntu14.04 with CUDA 8.0 + cuDNN 5.1.10 + Theano 0.8.2 + Lasagne 0.2.dev1.
1.2K
Linux enviroment for running salgan model.
Start an interactive session in the container by running:
sudo nvidia-docker run -it --entrypoint='bash' -w /home/ evamohe/salgan
For testing model inference:
cd /home/salgan/scripts THEANO_FLAGS=mode=FAST_RUN,device=gpu0,floatX=float32,lib.cnmem=0.5,optimizer_including=cudnn python 03-predict.py
Content type
Image
Digest
Size
2.7 GB
Last updated
about 8 years ago
docker pull evamohe/salgan