You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
507 B
Bash
21 lines
507 B
Bash
#!/bin/bash
|
|
|
|
# 安装构建工具
|
|
# Install build tools
|
|
conda install -c conda-forge gcc=14
|
|
conda install -c conda-forge gxx
|
|
conda install ffmpeg cmake
|
|
|
|
# 设置编译环境
|
|
# Set up build environment
|
|
export CMAKE_MAKE_PROGRAM="$CONDA_PREFIX/bin/cmake"
|
|
export CC="$CONDA_PREFIX/bin/gcc"
|
|
export CXX="$CONDA_PREFIX/bin/g++"
|
|
|
|
conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia
|
|
|
|
# 刷新环境
|
|
# Refresh environment
|
|
hash -r
|
|
|
|
pip install -r requirements.txt |