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:
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:
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?