Solved by peterjc123 in post #4 
              
              
                Please uninstall cpuonly in your conda environment. If torch.version.cuda returns none, then it means that you are using a CPU only binary.