Yes, you can do that if you are using CUDA 8.0. There’s an indention bug in 0.2, you can change the file C:\Users\wbruc\Anaconda3\lib\site-packages\torch\backends\cudnn\__init__.py. It can be easily found in the init function.