Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 1.57 KB

compile_from_source.md

File metadata and controls

20 lines (14 loc) · 1.57 KB
Raw

Compiling from source

Basic steps.

  1. make [target] where [target] is among cuda92, cuda10x, cuda110, cuda11x, cpuonly
  2. CUDA_VERSION=XXX python setup.py install

To run these steps you will need to have the nvcc compiler installed that comes with a CUDA installation. If you use anaconda (recommended) then you can figure out which version of CUDA you are using with PyTorch via the command conda list | grep cudatoolkit. Then you can install the nvcc compiler by downloading and installing the same CUDA version from the CUDA toolkit archive.

For your convenience, there is an installation script in the root directory that installs CUDA 11.1 locally and configures it automatically. After installing you should add the bin sub-directory to the $PATH variable to make the compiler visible to your system. To do this you can add this to your .bashrc by executing these commands:

echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64/" >> ~/.bashrc
echo "export PATH=$PATH:/usr/local/cuda/bin/" >> ~/.bashrc
source ~/.bashrc

By default, the Makefile will look at your CUDA_HOME environmental variable to find your CUDA version for compiling the library. If this path is not set it is inferred from the path of your nvcc compiler.

Either nvcc needs to be in path for the CUDA_HOME variable needs to be set to the CUDA directory root (e.g. /usr/local/cuda) in order for compilation to succeed

If you have problems compiling the library with these instructions from source, please open an issue.