TORCH CUDA issue / cellpose
0
0
Entering edit mode
4 months ago

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 | +-----------------------------------------------------------------------------+

cuda gpu torch • 202 views
ADD COMMENT
0
Entering edit mode

Maybe too late but you're on the wrong forum. Ask on forum.image.sc.

ADD REPLY

Login before adding your answer.

Traffic: 1593 users visited in the last hour
Help About
FAQ
Access RSS
API
Stats

Use of this site constitutes acceptance of our User Agreement and Privacy Policy.

Powered by the version 2.3.6