summaryrefslogtreecommitdiff
path: root/compile_from_source.md
diff options
context:
space:
mode:
authorTim Dettmers <TimDettmers@users.noreply.github.com>2022-07-18 09:51:37 -0700
committerGitHub <noreply@github.com>2022-07-18 09:51:37 -0700
commit4cd7ea62b2f51c68aacde2f62e7141765e476111 (patch)
tree548b2e77c62acd152330e898a6e17ea949a156d1 /compile_from_source.md
parent3418cd390e952a7752fb6b2544c25e25af7c0371 (diff)
parentfd750cd2370b3b12e216a9148b23aaae63a80989 (diff)
Merge pull request #3 from TimDettmers/cpuonly
Add a CPU-only build option
Diffstat (limited to 'compile_from_source.md')
-rw-r--r--compile_from_source.md4
1 files changed, 2 insertions, 2 deletions
diff --git a/compile_from_source.md b/compile_from_source.md
index c6a8b18..71b0c09 100644
--- a/compile_from_source.md
+++ b/compile_from_source.md
@@ -1,12 +1,12 @@
# Compiling from source
Basic steps.
-1. `make cudaXXX` where `cudaXXX` is among `cuda92, cuda10x, cuda110, cuda11x`
+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](https://developer.nvidia.com/cuda-toolkit-archive).
-For your convenience, there is a install script int he 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:
+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:
```bash
echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64/" >> ~/.bashrc
echo "export PATH=$PATH:/usr/local/cuda/bin/" >> ~/.bashrc