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

ppc64le nv meh

deepcrayon-v2.1
Jeff Moe 2023-11-08 13:03:13 -07:00
parent 880eb1f767
commit df96fa65fa
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@ -23,6 +23,7 @@ sudo apt install git build-essential libssl-dev zlib1g-dev \
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev \
libffi-dev liblzma-dev gcc-11 g++-11 libblis64-dev ninja-build \
libfftw3-dev libmpfr-dev protobuf-compiler protobuf-c-compiler \
libasmjit-dev \
python3-virtualenv python3-pip
```
@ -188,15 +189,16 @@ python setup.py build --cmake-only
# For amd64 CPU:
cmake build -DBLAS=BLIS -DTP_BUILD_PYTHON=ON
# For amd64 nvidia A6000 GPU (`sm_86`):
# For amd64 nvidia A6000 GPU (`sm_86`) XXX NON-FREE:
cmake build -DCUDAToolkit_INCLUDE_DIR=/usr/include -DBLAS=BLIS \
-DCUDA_SDK_ROOT_DIR=/usr -DENABLE_CUDA=ON -DTP_BUILD_PYTHON=ON
# For ppc64le CPU:
cmake build
cmake build -DUSE_NCCL=OFF
# For ppc64le testing nvidia A5000 GPU (`sm_86`):
cmake build -DCUDAToolkit_INCLUDE_DIR=/usr/include
cmake build -DCUDAToolkit_INCLUDE_DIR=/usr/include -DBLAS=BLIS \
-DCUDA_SDK_ROOT_DIR=/usr -DENABLE_CUDA=ON -DTP_BUILD_PYTHON=ON
# Make a wheel:
python setup.py bdist_wheel