forked from facebookresearch/detectron2
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Edited the Action file and the recipe
- Loading branch information
Showing
2 changed files
with
55 additions
and
37 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -9,49 +9,67 @@ jobs: | |
build_ubuntu: | ||
runs-on: ubuntu-latest | ||
steps: | ||
- name: Checkout Code | ||
uses: actions/checkout@v2 | ||
|
||
- name: Set GIT_TAG and GIT_COMMIT | ||
run: | | ||
echo "GIT_TAG=$(git describe --tags --abbrev=0)" >> $GITHUB_ENV | ||
echo "GIT_COMMIT=$(git rev-parse HEAD)" >> $GITHUB_ENV | ||
- name: Setup Miniconda | ||
id: setup-miniconda | ||
# You may pin to the exact commit or the version. | ||
# uses: conda-incubator/setup-miniconda@3b0f2504dd76ef23b6d31f291f4913fb60ab5ff3 | ||
uses: conda-incubator/[email protected] | ||
|
||
- name: Configure Miniconda | ||
run: | | ||
conda install --yes -n base conda-libmamba-solver conda-build conda-verify | ||
conda config --set solver libmamba | ||
- name: Setup CUDA | ||
- uses: Jimver/[email protected] | ||
run: conda install --yes -c nvidia -c conda-forge cudatoolkit=11.8 cudnn nvcc_win-64=11.8 cuda-tools=11.8 cuda-nvcc=11.8 | ||
id: cuda-toolkit | ||
with: | ||
cuda: '11.8.0' | ||
|
||
- name: Check CUDA Version | ||
- name: Conda Build and Publish | ||
run: | | ||
echo "Installed cuda version is: ${{ steps.cuda-toolkit.outputs.cuda }}" | ||
echo "Cuda install location: ${{ steps.cuda-toolkit.outputs.CUDA_PATH }}" | ||
nvcc -V | ||
- name: Checkout Code | ||
uses: actions/checkout@v2 | ||
|
||
- name: Conda Build and Publish on Ubuntu | ||
uses: fcakyon/[email protected] | ||
with: | ||
subdir: ./recipe # Sub-directory with conda recipe, optional, default is . | ||
anacondatoken: ${{ secrets.ANACONDA_TOKEN }} # Anaconda access token, optional | ||
platforms: 'linux' # Platforms to publish [osx/linux/win], optional, default is osx linux | ||
export TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX" | ||
export CUDA_HOME="/usr/miniconda3" | ||
conda build ./recipe -c pytorch -c nvidia -c conda-forge | ||
build_windows: | ||
runs-on: windows-latest | ||
steps: | ||
- name: Checkout Code | ||
uses: actions/checkout@v2 | ||
|
||
- name: Fetch all history for all tags and branches | ||
run: git fetch --unshallow | ||
|
||
- name: Set GIT_TAG and GIT_COMMIT | ||
run: | | ||
echo "GIT_TAG=$(git describe --tags --abbrev=0)" >> $GITHUB_ENV | ||
echo "GIT_COMMIT=$(git rev-parse HEAD)" >> $GITHUB_ENV | ||
- name: Setup Miniconda | ||
id: setup-miniconda | ||
# You may pin to the exact commit or the version. | ||
# uses: conda-incubator/setup-miniconda@3b0f2504dd76ef23b6d31f291f4913fb60ab5ff3 | ||
uses: conda-incubator/[email protected] | ||
|
||
- name: Configure Miniconda | ||
run: | | ||
conda install --yes -n base conda-libmamba-solver conda-build conda-verify | ||
conda config --set solver libmamba | ||
- name: Setup CUDA | ||
- uses: Jimver/[email protected] | ||
run: conda install --yes -c nvidia -c conda-forge cudatoolkit=11.8 cudnn nvcc_win-64=11.8 cuda-tools=11.8 cuda-nvcc=11.8 | ||
id: cuda-toolkit | ||
with: | ||
cuda: '11.8.0' | ||
|
||
- name: Check CUDA Version | ||
- name: Conda Build and Publish | ||
run: | | ||
echo "Installed cuda version is: ${{ steps.cuda-toolkit.outputs.cuda }}" | ||
echo "Cuda install location: ${{ steps.cuda-toolkit.outputs.CUDA_PATH }}" | ||
nvcc -V | ||
- name: Checkout Code | ||
uses: actions/checkout@v2 | ||
|
||
- name: Conda Build and Publish on Windows | ||
uses: fcakyon/[email protected] | ||
with: | ||
subdir: ./recipe # Sub-directory with conda recipe, optional, default is . | ||
anacondatoken: ${{ secrets.ANACONDA_TOKEN }} # Anaconda access token, optional | ||
platforms: 'win' # Platforms to publish [osx/linux/win], optional, default is osx linux | ||
$Env:TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX" | ||
$Env:CUDA_HOME="C:/miniconda3" | ||
conda build ./recipe -c pytorch -c nvidia -c conda-forge | ||
shell: powershell |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters