gpu version of pytorch not working on docker image #2010
Description
The docker image was created by running the command on the README file. My nvidia drivers work fine, and the cpu version of this works fine. But, running the classifier on the gpu as mentioned in http://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html gives the following error:
(pytorch-py35) root@eba92d25c1cd:~# nvidia-smi
Fri Jul 7 19:24:32 2017
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.39 Driver Version: 375.39 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 TITAN X (Pascal) Off | 0000:04:00.0 On | N/A |
| 23% 26C P8 16W / 250W | 1331MiB / 12189MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
| 1 TITAN X (Pascal) Off | 0000:06:00.0 Off | N/A |
| 23% 29C P8 15W / 250W | 4MiB / 12189MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
+-----------------------------------------------------------------------------+
(pytorch-py35) root@eba92d25c1cd:~# nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61
(pytorch-py35) root@eba92d25c1cd:~# python classifier.py
Files already downloaded and verified
Files already downloaded and verified
Traceback (most recent call last):
File "classifier.py", line 45, in <module>
net.cuda()
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 147, in cuda
return self._apply(lambda t: t.cuda(device_id))
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 118, in _apply
module._apply(fn)
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 124, in _apply
param.data = fn(param.data)
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 147, in <lambda>
return self._apply(lambda t: t.cuda(device_id))
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/_utils.py", line 65, in _cuda
return new_type(self.size()).copy_(self, async)
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/cuda/__init__.py", line 272, in __new__
_lazy_init()
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/cuda/__init__.py", line 84, in _lazy_init
_check_driver()
File "/opt/conda/envs/pytorch-py35/lib/python3.5/site-packages/torch/cuda/__init__.py", line 51, in _check_driver
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled