deepcrayon
/
pytorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Activity

More build deps

deepcrayon-v2.1
Jeff Moe 2023-11-08 12:06:08 -07:00
parent 92ca6b5192
commit 0cb15d1569
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -21,7 +21,8 @@ Perhaps this and more:
sudo apt install git build-essential libssl-dev zlib1g-dev \
libbz2-dev libreadline-dev libsqlite3-dev curl ccache \
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev \
libffi-dev liblzma-dev gcc-11 \
libffi-dev liblzma-dev gcc-11 g++-11 libblis64-dev ninja-build \
libfftw3-dev libmpfr-dev protobuf-compiler protobuf-c-compiler \
python3-virtualenv python3-pip
```
@ -184,6 +185,9 @@ python setup.py build --cmake-only
# Use `ccmake` instead of `cmake` if you want to configure further.
#
# For amd64 CPU:
cmake build -DBLAS=BLIS -DTP_BUILD_PYTHON=ON
# For amd64 nvidia A6000 GPU (`sm_86`):
cmake build -DCUDAToolkit_INCLUDE_DIR=/usr/include -DBLAS=BLIS \
-DCUDA_SDK_ROOT_DIR=/usr -DENABLE_CUDA=ON -DTP_BUILD_PYTHON=ON