Build notes, make wheel

deepcrayon-v2.1
Jeff Moe 2023-11-08 10:45:10 -07:00
parent ef45776b7e
commit ad87bd083f
1 changed files with 8 additions and 5 deletions

View File

@ -152,23 +152,23 @@ Now actually build.
From System76, but came with A5000.
Has non-free Debian nvidia junk installed.
```
#!/bin/bash
This rebuilds from scratch.
```
export PYTHONVER=3.10
export TORCHVER=2.1.0
export TORCHVER=2.1.0-dc1
export GCCVER=11
export CMAKE_C_COMPILER=/usr/lib/ccache/gcc-${GCCVER}
export CMAKE_CXX_COMPILER=/usr/lib/ccache/g++-${GCCVER}
cd ~/devel/pytorch/pytorch
cd ~/devel/deepcrayon/pytorch # or wherever repo is
source deactivate
rm -rf venv
rm .python-version
rm -rf build
mkdir -p build
git checkout main
git checkout deepcrayon-v2.1
git clean -ff
git reset --hard HEAD
git clean -ff
@ -190,6 +190,9 @@ python setup.py build --cmake-only
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
# or:
python setup.py install
```