seuliufeng / DeepSBIR

Code for paper Sketch Me That Shoe
61 stars 27 forks source link

cuda cudnn version #4

Open LuPangCute opened 6 years ago

LuPangCute commented 6 years ago

which version of your cuda and cudnn

umdreamer commented 5 years ago

I used cuda 9.0 and cudnn 7.0. You have to replace the files in caffesbir/src/caffe/layers/cuda.cu and cuda_.cpp with the latest official caffe.