search

jessieren / DeepVirFinder

Identifying viruses from metagenomic data by deep learning
Other
116 stars 32 forks source link

Training on Windows #13

Closed andrewjmc closed 4 years ago

andrewjmc commented 4 years ago

Total GPU beginner question here, apologies, but grateful for pointers.

Whilst I will likely ultimately use DeepVirFinder on HPC, I am testing it out on a Windows box with NVidia GPU (Quadro K620) as I can iterate quicker. I installed the latest NVidia CUDA and cuDNN drivers following their instructions. I use the environment variable as specified in the training test: THEANO_FLAGS='mode=FAST_RUN,device=cuda0,floatX=float32,GPUARRAY_CUDA_VERSION=80'

However, I cannot find the Windows equivalent for:

source /<path_to_cuda_setup>/setup.sh
source /<path_to_cuDNN_setup>/setup.sh

And it seems to run with CPU regardless of what I do.

Do I need specific versions of CUDA/cuDNN? And do I do anything special to let python use them on Windows?

andrewjmc commented 4 years ago

I'll close this issue as I've subsequently got it working on Linux. Still not sure what the setup.sh is about.

Best wishes,

Andrew