I tried running your code in my system. But I ran into some error when executing the command python example.py as:
Traceback (most recent call last): File "example.py", line 1, in <module> import torch File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torch/__init__.py", line 102, in <module> from torch._C import * ImportError: /home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torch/lib/libtorch.so.1: undefined symbol: nvrtcGetProgramLogSize
Please address this issue. Waiting for a reply.
I tried running your code in my system. But I ran into some error when executing the command
python example.pyas:Traceback (most recent call last): File "example.py", line 1, in <module> import torch File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torch/__init__.py", line 102, in <module> from torch._C import * ImportError: /home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torch/lib/libtorch.so.1: undefined symbol: nvrtcGetProgramLogSizePlease address this issue. Waiting for a reply.