Zenn
Open5

PyTorch v1.12.0 + grid_smple (grid_sampler) opset=16 + CUDA11.6 + cuDNN8.4 のビルド試行

PyTorch
CUDA
PINTOPINTO
  • PyTorch v1.12.0
docker run --gpus all -it --rm \
-v `pwd`:/home/user/workdir \
ghcr.io/pinto0309/openvino2tensorflow:latest

git clone --recursive https://github.com/pytorch/pytorch \
&& cd pytorch
git checkout e68686bb05f6df4769dab638d70aa320f3666673

sudo sed -i -e \
"/^#ifndef THRUST_IGNORE_CUB_VERSION_CHECK$/i #define THRUST_IGNORE_CUB_VERSION_CHECK" \
/usr/local/cuda/targets/x86_64-linux/include/thrust/system/cuda/config.h

pip3 install -r requirements.txt \
&& USE_NCCL=OFF python3 setup.py build \
&& python3 setup.py bdist_wheel

find . -name "*.whl"
./dist/torch-1.12.0a0+gite36d25f-cp38-cp38-linux_x86_64.whl

pip install ./dist/torch-1.12.0a0+gite36d25f-cp38-cp38-linux_x86_64.whl
PINTOPINTO
  • TorchVision v0.13.0
git clone https://github.com/pytorch/vision \
&& cd vision
git checkout ecbff88a1ad605bf04d6c44862e93dde2fdbfc84

git submodule update --init --recursive \
&& python3 setup.py build \
&& python3 setup.py bdist_wheel

find . -name "*.whl"
./dist/torchvision-0.13.0a0+ecbff88-cp38-cp38-linux_x86_64.whl

pip install ./dist/torchvision-0.13.0a0+ecbff88-cp38-cp38-linux_x86_64.whl
PINTOPINTO
  • TorchAudio v0.12.0
git clone https://github.com/pytorch/audio \
&& cd audio
git checkout a71e3a401e4a68d23900ff7d3d0437676392fa19

git submodule update --init --recursive \
&& sudo apt update \
&& sudo apt install -y sox libsox-dev \
&& python3 setup.py build \
&& python3 setup.py bdist_wheel

find . -name "*.whl"
./dist/torchaudio-0.12.0a0+a71e3a4-cp38-cp38-linux_x86_64.whl