1.Vulcan

https://vulkan.lunarg.com/sdk/home

tar -xvzf vulkansdk-linux-x86_64-1.3.239.0.tar.gz
cd 1.3.239.0
./vulkansdk -j 20

2.pytorch

git clone https://github.com/pytorch/pytorch.git
cd pytorch
git submodule sync
git submodule update --init --recursive
git checkout release/0.13

USE_VULKAN=1 USE_VULKAN_SHADERC_RUNTIME=1 USE_VULKAN_WRAPPER=0  python setup.py develop

rm /home/smallsung/anaconda3/lib/libstdc++.so.6
ln -s /usr/lib/x86_64-linux-gnu/libstdc++.so.6 /home/smallsung/anaconda3/lib/libstdc++.so.6

python -c "import torch"

3.torchaudio

conda install cmake ninja pkg-config
conda install -c conda-forge ffmpeg
git clone https://github.com/pytorch/audio
cd audio
git checkout release/0.13
USE_FFMPEG=1 python setup.py develop

4.timm

git clone https://github.com/rwightman/pytorch-image-models.git
cd pytorch-image-models
pip install -e . --no-deps

5.Android

cd pytorch
ANDROID_NDK=</home/smallsung/Android/Sdk/ndk/21.4.7075529/> ANDROID_ABI=arm64-v8a USE_VULKAN=1 bash ./scripts/build_android.sh
ANDROID_NDK=</home/smallsung/Android/Sdk/ndk/21.4.7075529/> ANDROID_HOME=</home/smallsung/Android/Sdk> JAVA_HOME=</usr/local/android-studio/jbr> USE_VULKAN=1 BUILD_LITE_INTERPRETER=0 bash ./scripts/build_pytorch_android.sh
Last modification:July 11, 2023
如果觉得我的文章对你有用,请随意赞赏