Hi,inspite of upgrading torch to torch==1.9.1 and torchvision=0.10.1
i am facing this issue
Traceback (most recent call last):
File “train.py”, line 357, in 
main()
File “train.py”, line 112, in main
model = Network(args.init_channels, num_domains, args.layers, criterion)
File “/voyager-volume/code_1_test/code_1_test/code_1_test/code_1_test/code_1_test/model_search.py”, line 95, in init
self._initialize_alphas()
File “/voyager-volume/code_1_test/code_1_test/code_1_test/code_1_test/code_1_test/model_search.py”, line 124, in _initialize_alphas
self.alphas_normal = Variable(1e-3*torch.randn(k, num_ops).cuda(), requires_grad=True)
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.