Ok, that’s awesome!

My torch.__version__ is 0.3.0.post4

I installed using conda (miniconda 3) on linux.

cat /usr/local/cuda/version.txt -> CUDA Version 8.0.61


nvidia-smi :

Fri Jan 19 16:34:41 2018       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.26                 Driver Version: 375.26                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Quadro M2000        Off  | 0000:03:00.0      On |                  N/A |
| 58%   49C    P0    25W /  75W |   2181MiB /  4036MiB |      1%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|    0      1298    G   /opt/viber/Viber                                49MiB |
|    0      1984    G   /usr/lib/xorg/Xorg                               0MiB |
|    0      2742    G   /usr/lib/xorg/Xorg                             282MiB |
|    0     19032    G   /usr/bin/gnome-shell                           225MiB |
|    0     19704    G   /usr/lib/firefox/firefox                         1MiB |
|    0     21829    C   /usr/lib/libreoffice/program/soffice.bin        31MiB |
|    0     26772    G   /usr/lib/firefox/firefox                       162MiB |
|    0     26774    G   /usr/lib/firefox/firefox                        62MiB |
|    0     26780    G   /usr/lib/firefox/firefox                         1MiB |
|    0     29663    G   ~/miniconda3/bin/python                         41MiB |
|    0     29767    C   ~/miniconda3/bin/python                       1313MiB |
+-----------------------------------------------------------------------------+