CUDA error: all CUDA-capable devices are busy or unavailable about bringing-old-photos-back-to-life HOT 3 CLOSED

Maybe you can do the simplest test in python:

import torch
print(torch.cuda.is_available())

If cuda cannot be detected by torch, maybe you can try

import torch
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
print(torch.cuda.is_available())

from bringing-old-photos-back-to-life.

Maybe you can do the simplest test in python:

import torch
print(torch.cuda.is_available())

If cuda cannot be detected by torch, maybe you can try

import torch
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
print(torch.cuda.is_available())

thank for the replay，my cuda looks normal

(venv) dash@dash-KVM:~/Bringing-Old-Photos-Back-to-Life-master$ python
Python 3.6.9 (default, Oct 8 2020, 12:12:24)
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.

import torch
print(torch.cuda.is_available())
True
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
print(torch.cuda.is_available())
True

from bringing-old-photos-back-to-life.

Maybe you can try to run with --GPU -1. Or you can raise the issue in the official PyTorch repo as it seems the problem of PyTorch installation.

from bringing-old-photos-back-to-life.