Hi there,
I am using pyroc to run my code. I install anaconda then i add these variable
export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:/usr/local/cuda-7.5/lib:$LD_LIBRARY_PATH
export CPATH=/usr/local/cuDNN-6.0.21/include:$CPATH
export LD_LIBRARY_PATH=/usr/local/cuDNN-6.0.21/lib64:$LD_LIBRARY_PATH
export LIBRARY_PATH=/usr/local/cuDNN-6.0.21/lib64:$LIBRARY_PATH
then my code is work but slow Iam not sure if i use GPU pytorch