PyTorch with CUDA 9.1 is super slow compared to 9.0

Ah, thanks @smth ! I must say it’s a very weird case, I never got a warning when installing/importing it. And despite the fact that it detects the CUDA 9.1 lib, it does run on CPU (0% utilization when I check via nvidia-smi). When I install the CUDA 8.0 PyTorch version (also via conda), it works just fine though with 100% utilization.