diff --git a/main.py b/main.py index 392242902..05eb31c7a 100644 --- a/main.py +++ b/main.py @@ -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))