Setting up your GPU
How to speed up cellfinder by using your GPU
cellfinder will run quite happily on your CPU, but the machine learning parts (classifying cell candidates as cells or artefacts, and Training the network) run much faster using GPU.
- A relatively modern Windows or Linux based machine (unfortunately GPU acceleration on macOS is not supported).
- An NVIDIA GPU with CUDA Compute Capability of 3.5 or higher (see the list here) with at least 6GB memory. Basically any relatively expensive NVIDIA GPU released in the last 5 years should be OK. So far, we've tested:
- GTX 1070
- GTX 1080
- TITAN GTX
- Quadro P5000
- Quadro RTX 5000
- RTX 2080
- RTX 2080 (laptop)
- TITAN RTX
- Someone who has the admin password and (ideally) knows what they are doing to install things.
These instructions will vary somewhat between operating systems, and on how your machine is set up (and how much control your institute will let you have).
The first thing you definitely need are the drivers for your GPU, which can be downloaded here. Hopefully, these will have been installed when your machine was set up, but for GPU support in cellfinder, you will need version 450.x or greater.
CUDA and cuDNN are not too hard to install, but sometimes other software on your machine relies on different versions. It is possible to switch between the two, and it is easier if you are using conda (see here). However, I recommend that you use conda and install CUDA 11.2 and cuDNN 8.1 via conda. On linux you can run:
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
This method is easier, and also doesn't require any admin rights (useful on a cluster or shared machine).
Last modified 4mo ago