I will try to explain my issue with all details:
I want to used Cellpose on an HPC servers. I created a conda env by using the environement.yml file provide by the github of cellpose. All seems to work well. But now i want to used the GPU version of cellpose to speed up the analysis. I follow the instruction on their github :
pip uninstall torch
conda install pytorch cudatoolkit=10.2 -c pytorch
But when i try to used the GPU i obtained this error: TORCH CUDA version not installed/working.
How could i manage this issue?
I precise that we have a GPU node in our HPC cluster. When i run the command nvidia-smi i obtained these informations:
+-----------------------------------------------------------------------------+ | NVIDIA-SMI 418.39 Driver Version: 418.39 CUDA Version: 10.1 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 Tesla K80 On | 00000000:06:00.0 Off | 0 | | N/A 26C P8 26W / 149W | 0MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 1 Tesla K80 On | 00000000:07:00.0 Off | 0 | | N/A 31C P8 28W / 149W | 0MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 2 Tesla K80 On | 00000000:0A:00.0 Off | 0 | | N/A 25C P8 26W / 149W | 0MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 3 Tesla K80 On | 00000000:0B:00.0 Off | 0 | | N/A 30C P8 29W / 149W | 0MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 4 Tesla K80 On | 00000000:86:00.0 Off | 0 | | N/A 27C P8 26W / 149W | 0MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 5 Tesla K80 On | 00000000:87:00.0 Off | 0 | | N/A 32C P8 28W / 149W | 11MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 6 Tesla K80 On | 00000000:8A:00.0 Off | 0 | | N/A 27C P8 26W / 149W | 11MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 7 Tesla K80 On | 00000000:8B:00.0 Off | 0 | | N/A 31C P8 29W / 149W | 0MiB / 11441MiB | 0% Default | +-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+ | Processes: GPU Memory | | GPU PID Type Process name Usage | |=============================================================================| | No running processes found | +-----------------------------------------------------------------------------+
Maybe too late but you're on the wrong forum. Ask on forum.image.sc.