env.py

import spaces
import os
import torch

@spaces.GPU(duration=5)
def check():
    print(os.system("nvidia-smi"))
    print(torch.version.cuda)

check()