check if my torch using GPU

0 views
Skip to first unread message

ebtesam h

unread,
Nov 8, 2017, 8:20:16 AM11/8/17
to conda - Public
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
Reply all
Reply to author
Forward
0 new messages