https://gist.github.com/victor-shepardson/5b3d3087dc2b4817b9bffdb8e87a57c4
I'm using Ubuntu 16.04 with a GTX 1060
I had to use the .run to get CUDA 8 back when Pascal was new, for some reason lost to time. The apt package probably works fine these days.
set PATH, CUDA_HOME, LD_LIBRARY_PATH appropriately.
download for python3.6, run the script.
https://www.continuum.io/downloads
The following assumes you set Anaconda as the default python; if not, replace python
and pip
with e.g. ~/anaconda3/bin/python
sudo pip install glumpy
http://glumpy.readthedocs.io/en/latest/index.html
(I had to because of old AMD CPU, otherwise conda package is probably fine)
https://github.com/pytorch/pytorch#from-source
Necessary for opengl interop, also possibly for some CUDA locations.
clone or untar pycuda, cd in
get root with current environment: sudo -E su -p
python configure.py --cuda-root=$CUDA_HOME --cuda-enable-gl
python setup.py make
python setup.py install