Skip to content

pytorch installation problem #2177

@shwoo1219

Description

@shwoo1219

Hello

In order to follow up with the current state of the pytorch version in github, I followed the below setup.

Firstly, I build the image with cuda 8.0-ubuntu 14.04

On top of this image, I made a container and installed anaconda and opencv.

bash Anaconda3-4.4.0-Linux-x86_64.sh
conda update conda
conda install opencv

After then, I cloned this github page and followed the instructions provided.

git clone https://github.com/pytorch/pytorch.git
export CMAKE_PREFIX_PATH=[anaconda root directory]
conda install numpy pyyaml mkl setuptools cmake gcc cffi
conda install -c soumith magma-cuda80
python setup.py install
conda install torchvision

After the above installation is finished, I tried to run my code but I got the following error....

Loading base network...
Traceback (most recent call last):
File "train_deform.py", line 93, in
net.cuda()
File "/root/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 147, in cuda
return self._apply(lambda t: t.cuda(device_id))
File "/root/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 118, in _apply
module._apply(fn)
File "/root/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 124, in _apply
param.data = fn(param.data)
File "/root/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 147, in
return self._apply(lambda t: t.cuda(device_id))
File "/root/anaconda3/lib/python3.6/site-packages/torch/_utils.py", line 65, in cuda
return new_type(self.size()).copy
(self, async)
File "/root/anaconda3/lib/python3.6/site-packages/torch/cuda/init.py", line 272, in new
_lazy_init()
File "/root/anaconda3/lib/python3.6/site-packages/torch/cuda/init.py", line 84, in _lazy_init
_check_driver()
File "/root/anaconda3/lib/python3.6/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

I also checked nvidia-smi and nvcc --version in order to check whether GPU and CUDA is well specified, However there was no problem with in it.

I cant find the problem. Since I just installed magma-cuda80 right before the installing setup.py. What is the problem ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions