OmiEmbed icon indicating copy to clipboard operation
OmiEmbed copied to clipboard

Published 20 hours ago • verified publisher icon zhangxiaoyu11

Code Minor fixes

Open daniocionini opened this issue 2 years ago • 0 comments

Modified code at 206 of basic_params.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

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

daniocionini avatar May 05 '22 19:05 daniocionini