Open daniocionini opened 2 years ago
Modified code at 206 of _basicparams.py as usage of torch.cud.set_device(device) is discouraged in favor of torch.cuda.device(device) https://pytorch.org/docs/stable/generated/torch.cuda.set_device.html
torch.cud.set_device(device)
torch.cuda.device(device)
Minor fix to the requirements.txt doc. PyTorch CUDA wont run if the graphics card isn't listed as one of those https://en.wikipedia.org/wiki/CUDA#GPUs_supported
requirements.txt
Modified code at 206 of _basicparams.py as usage of
torch.cud.set_device(device)is discouraged in favor oftorch.cuda.device(device)https://pytorch.org/docs/stable/generated/torch.cuda.set_device.htmlMinor fix to the
requirements.txtdoc. PyTorch CUDA wont run if the graphics card isn't listed as one of those https://en.wikipedia.org/wiki/CUDA#GPUs_supported