deleted accidental if

This commit is contained in:
Bratzmeister 2024-11-11 21:00:33 +01:00 committed by GitHub
parent 4c435fddb5
commit 832b290199
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -70,7 +70,6 @@ if os.name == "nt":
if __name__ == "__main__":
if args.cuda_device is not None:
if
os.environ['CUDA_VISIBLE_DEVICES'] = str(args.cuda_device)
os.environ['HIP_VISIBLE_DEVICES'] = str(args.cuda_device)
logging.info("Set cuda device to: {}".format(args.cuda_device))