So I think I figure it out. It turns out that it is because I didn’t reboot the system after installing pytorch using conda.