CUDA_VISIBLE_DEVICES=0 python3 main.py
import torch
import torchvision
print(torch.cuda.is_available())
print(torch.__version__)
CUDA_VISIBLE_DEVICES=0 python3 main.py
import torch
import torchvision
print(torch.cuda.is_available())
print(torch.__version__)