PyTorch v1.12.1 + CUDA11.7 のビルド試行

docker run --rm -it --gpus all \
-v `pwd`:/home/user/workdir \
ghcr.io/pinto0309/openvino2tensorflow:base.11.7.1-cudnn8-tf2.10.0-trt8.4.3-openvino2022.1.0

cd /home/user/workdir

git clone -b v1.12.1 --recursive https://github.com/pytorch/pytorch \
&& cd pytorch

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+git664058f-cp38-cp38-linux_x86_64.whl

pip install ./dist/torch-1.12.0a0+git664058f-cp38-cp38-linux_x86_64.whl

git clone -b v0.13.1 https://github.com/pytorch/vision \
&& cd vision

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

find . -name "*.whl"
./dist/torchvision-0.13.1a0+bddbd7e-cp38-cp38-linux_x86_64.whl

pip install ./dist/torchvision-0.13.1a0+bddbd7e-cp38-cp38-linux_x86_64.whl

git clone -b v0.12.1 https://github.com/pytorch/audio \
&& cd audio

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.1+58da317-cp38-cp38-linux_x86_64.whl

pip install ./dist/torchaudio-0.12.1+58da317-cp38-cp38-linux_x86_64.whl