Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nerfacc
Commits
14398229
Unverified
Commit
14398229
authored
Jan 31, 2023
by
Ruilong Li(李瑞龙)
Committed by
GitHub
Jan 31, 2023
Browse files
Workflows to build wheel (#151)
pre-built wheels
parent
e9dd9da6
Changes
41
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
393 additions
and
4 deletions
+393
-4
.github/workflows/building.yml
.github/workflows/building.yml
+170
-0
.github/workflows/code_checks.yml
.github/workflows/code_checks.yml
+4
-4
.github/workflows/cuda/cu101-Linux-env.sh
.github/workflows/cuda/cu101-Linux-env.sh
+8
-0
.github/workflows/cuda/cu101-Linux.sh
.github/workflows/cuda/cu101-Linux.sh
+15
-0
.github/workflows/cuda/cu101-Windows-env.sh
.github/workflows/cuda/cu101-Windows-env.sh
+8
-0
.github/workflows/cuda/cu101-Windows.sh
.github/workflows/cuda/cu101-Windows.sh
+18
-0
.github/workflows/cuda/cu102-Linux-env.sh
.github/workflows/cuda/cu102-Linux-env.sh
+8
-0
.github/workflows/cuda/cu102-Linux.sh
.github/workflows/cuda/cu102-Linux.sh
+15
-0
.github/workflows/cuda/cu102-Windows-env.sh
.github/workflows/cuda/cu102-Windows-env.sh
+8
-0
.github/workflows/cuda/cu102-Windows.sh
.github/workflows/cuda/cu102-Windows.sh
+18
-0
.github/workflows/cuda/cu111-Linux-env.sh
.github/workflows/cuda/cu111-Linux-env.sh
+8
-0
.github/workflows/cuda/cu111-Linux.sh
.github/workflows/cuda/cu111-Linux.sh
+15
-0
.github/workflows/cuda/cu111-Windows-env.sh
.github/workflows/cuda/cu111-Windows-env.sh
+8
-0
.github/workflows/cuda/cu111-Windows.sh
.github/workflows/cuda/cu111-Windows.sh
+18
-0
.github/workflows/cuda/cu113-Linux-env.sh
.github/workflows/cuda/cu113-Linux-env.sh
+8
-0
.github/workflows/cuda/cu113-Linux.sh
.github/workflows/cuda/cu113-Linux.sh
+15
-0
.github/workflows/cuda/cu113-Windows-env.sh
.github/workflows/cuda/cu113-Windows-env.sh
+8
-0
.github/workflows/cuda/cu113-Windows.sh
.github/workflows/cuda/cu113-Windows.sh
+18
-0
.github/workflows/cuda/cu115-Linux-env.sh
.github/workflows/cuda/cu115-Linux-env.sh
+8
-0
.github/workflows/cuda/cu115-Linux.sh
.github/workflows/cuda/cu115-Linux.sh
+15
-0
No files found.
.github/workflows/building.yml
0 → 100644
View file @
14398229
name
:
Building Wheels
on
:
[
workflow_dispatch
]
jobs
:
wheel
:
runs-on
:
${{ matrix.os }}
environment
:
production
strategy
:
fail-fast
:
false
matrix
:
os
:
[
ubuntu-18.04
,
windows-2019
]
python-version
:
[
'
3.7'
,
'
3.8'
,
'
3.9'
]
torch-version
:
[
1.10.0
,
1.11.0
,
1.12.0
,
1.13.0
]
cuda-version
:
[
'
cu102'
,
'
cu113'
,
'
cu116'
,
'
cu117'
]
# os: [ubuntu-18.04]
# python-version: ['3.9']
# torch-version: [1.10.0]
# cuda-version: ['cu102']
exclude
:
-
torch-version
:
1.10.0
cuda-version
:
'
cu116'
-
torch-version
:
1.10.0
cuda-version
:
'
cu117'
-
torch-version
:
1.11.0
cuda-version
:
'
cu116'
-
torch-version
:
1.11.0
cuda-version
:
'
cu117'
-
torch-version
:
1.12.0
cuda-version
:
'
cu117'
-
torch-version
:
1.13.0
cuda-version
:
'
cu102'
-
torch-version
:
1.13.0
cuda-version
:
'
cu113'
-
os
:
windows-2019
torch-version
:
1.11.0
cuda-version
:
'
cu102'
-
os
:
windows-2019
torch-version
:
1.12.0
cuda-version
:
'
cu102'
# - os: macos-10.15
# cuda-version: 'cu102'
# - os: macos-10.15
# cuda-version: 'cu113'
# - os: macos-10.15
# cuda-version: 'cu116'
# - os: macos-10.15
# cuda-version: 'cu117'
steps
:
-
uses
:
actions/checkout@v2
-
name
:
Set up Python ${{ matrix.python-version }}
uses
:
actions/setup-python@v2
with
:
python-version
:
${{ matrix.python-version }}
-
name
:
Upgrade pip
run
:
|
pip install --upgrade setuptools
pip install ninja
-
name
:
Free up disk space
if
:
${{ runner.os == 'Linux' }}
run
:
|
sudo rm -rf /usr/share/dotnet
-
name
:
Install CUDA ${{ matrix.cuda-version }}
if
:
${{ matrix.cuda-version != 'cpu' }}
run
:
|
bash .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}.sh
-
name
:
Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
run
:
|
pip install torch==${{ matrix.torch-version }} --extra-index-url https://download.pytorch.org/whl/${{ matrix.cuda-version }}
python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)"
python -c "import torch; print('CUDA Available:', torch.cuda.is_available())"
-
name
:
Patch PyTorch static constexpr on Windows
if
:
${{ runner.os == 'Windows' }}
run
:
|
Torch_DIR=`python -c 'import os; import torch; print(os.path.dirname(torch.__file__))'`
sed -i '31,38c\
TORCH_API void lazy_init_num_threads();' ${Torch_DIR}/include/ATen/Parallel.h
shell
:
bash
-
name
:
Set version
if
:
${{ runner.os != 'macOS' }}
run
:
|
VERSION=`sed -n 's/^__version__ = "\(.*\)"/\1/p' nerfacc/version.py`
TORCH_VERSION=`echo "pt${{ matrix.torch-version }}" | sed "s/..$//" | sed "s/\.//g"`
CUDA_VERSION=`echo ${{ matrix.cuda-version }}`
echo "New version name: $VERSION+$TORCH_VERSION$CUDA_VERSION"
sed -i "s/$VERSION/$VERSION+$TORCH_VERSION$CUDA_VERSION/" nerfacc/version.py
shell
:
bash
-
name
:
Install main package for CPU
if
:
${{ matrix.cuda-version == 'cpu' }}
run
:
|
FORCE_ONLY_CPU=1 pip install -e .
shell
:
bash
-
name
:
Install main package for GPU
if
:
${{ matrix.cuda-version != 'cpu' }}
run
:
|
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
pip install .
shell
:
bash
-
name
:
Test installation
run
:
|
python -c "import nerfacc; print('nerfacc:', nerfacc.__version__)"
-
name
:
Build wheel
run
:
|
pip install wheel
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
python setup.py bdist_wheel --dist-dir=dist
shell
:
bash
-
name
:
Configure AWS
uses
:
aws-actions/configure-aws-credentials@v1
with
:
aws-access-key-id
:
${{ secrets.AWS_ACCESS_KEY_ID }}
aws-secret-access-key
:
${{ secrets.AWS_SECRET_ACCESS_KEY }}
aws-region
:
us-west-2
-
name
:
Upload wheel
run
:
|
aws s3 sync dist s3://nerfacc-bucket/whl/torch-${{ matrix.torch-version }}_${{ matrix.cuda-version }} --grants read=uri=http://acs.amazonaws.com/groups/global/AllUsers
update_aws_listing
:
needs
:
[
wheel
]
runs-on
:
ubuntu-latest
environment
:
production
steps
:
-
uses
:
actions/checkout@v2
-
name
:
Set up Python
uses
:
actions/setup-python@v2
with
:
python-version
:
3.9
-
name
:
Upgrade pip
run
:
|
pip install --upgrade setuptools
pip install boto3
-
name
:
Configure AWS
uses
:
aws-actions/configure-aws-credentials@v1
with
:
aws-access-key-id
:
${{ secrets.AWS_ACCESS_KEY_ID }}
aws-secret-access-key
:
${{ secrets.AWS_SECRET_ACCESS_KEY }}
aws-region
:
us-west-2
-
name
:
Update AWS listing
run
:
|
python scripts/run_aws_listing.py \
--access_key_id=${{ secrets.AWS_ACCESS_KEY_ID }} \
--secret_access_key=${{ secrets.AWS_SECRET_ACCESS_KEY }} \
--bucket="nerfacc-bucket" \
--region="us-west-2"
.github/workflows/code_checks.yml
View file @
14398229
...
@@ -25,11 +25,11 @@ jobs:
...
@@ -25,11 +25,11 @@ jobs:
key
:
${{ env.pythonLocation }}-${{ hashFiles('pyproject.toml') }}
key
:
${{ env.pythonLocation }}-${{ hashFiles('pyproject.toml') }}
-
name
:
Install dependencies
-
name
:
Install dependencies
run
:
|
run
:
|
pip install
--upgrade --upgrade-strategy eager -e .[dev]
pip install
isort black
-
name
:
Run isort
-
name
:
Run isort
run
:
isort docs/ nerfacc/ scripts/ examples/ tests/ --profile black --skip examples/pycolmap --check
run
:
isort docs/ nerfacc/ scripts/ examples/ tests/ --profile black --skip examples/pycolmap
--line-length 80
--check
-
name
:
Run Black
-
name
:
Run Black
run
:
black docs/ nerfacc/ scripts/ examples/ tests/ --exclude examples/pycolmap --check
run
:
black docs/ nerfacc/ scripts/ examples/ tests/ --exclude examples/pycolmap
--line-length 80
--check
# - name: Python Pylint
# - name: Python Pylint
# run: |
# run: |
# pylint nerfacc/ tests/ scripts/ examples/
# pylint nerfacc/ tests/ scripts/ examples/
\ No newline at end of file
.github/workflows/cuda/cu101-Linux-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/usr/local/cuda-10.1
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5"
.github/workflows/cuda/cu101-Linux.sh
0 → 100755
View file @
14398229
#!/bin/bash
OS
=
ubuntu1804
wget
-nv
https://developer.download.nvidia.com/compute/cuda/repos/
${
OS
}
/x86_64/cuda-
${
OS
}
.pin
sudo mv
cuda-
${
OS
}
.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
-nv
https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-
${
OS
}
-10-1-local-10
.1.243-418.87.00_1.0-1_amd64.deb
sudo
dpkg
-i
cuda-repo-
${
OS
}
-10-1-local-10
.1.243-418.87.00_1.0-1_amd64.deb
sudo
apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
sudo
apt-get
-qq
update
sudo
apt
install
-y
cuda-nvcc-10-1 cuda-libraries-dev-10-1
sudo
apt clean
rm
-f
https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-
${
OS
}
-10-1-local-10
.1.243-418.87.00_1.0-1_amd64.deb
.github/workflows/cuda/cu101-Windows-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/c/Program
\
Files/NVIDIA
\
GPU
\
Computing
\
Toolkit/CUDA/v10.1
PATH
=
${
CUDA_HOME
}
/bin:
$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5"
.github/workflows/cuda/cu101-Windows.sh
0 → 100755
View file @
14398229
#!/bin/bash
# Install NVIDIA drivers, see:
# https://github.com/pytorch/vision/blob/master/packaging/windows/internal/cuda_install.bat#L99-L102
curl
-k
-L
"https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download"
--output
"/tmp/gpu_driver_dlls.zip"
7z x
"/tmp/gpu_driver_dlls.zip"
-o
"/c/Windows/System32"
export
CUDA_SHORT
=
10.1
export
CUDA_URL
=
https://developer.download.nvidia.com/compute/cuda/
${
CUDA_SHORT
}
/Prod/local_installers/
export
CUDA_FILE
=
cuda_
${
CUDA_SHORT
}
.243_426.00_win10.exe
# Install CUDA:
curl
-k
-L
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
--output
"
${
CUDA_FILE
}
"
echo
""
echo
"Installing from
${
CUDA_FILE
}
..."
PowerShell
-Command
"Start-Process -FilePath
\"
${
CUDA_FILE
}
\"
-ArgumentList
\"
-s nvcc_
${
CUDA_SHORT
}
cuobjdump_
${
CUDA_SHORT
}
nvprune_
${
CUDA_SHORT
}
cupti_
${
CUDA_SHORT
}
cublas_dev_
${
CUDA_SHORT
}
cudart_
${
CUDA_SHORT
}
cufft_dev_
${
CUDA_SHORT
}
curand_dev_
${
CUDA_SHORT
}
cusolver_dev_
${
CUDA_SHORT
}
cusparse_dev_
${
CUDA_SHORT
}
npp_dev_
${
CUDA_SHORT
}
nvrtc_dev_
${
CUDA_SHORT
}
nvml_dev_
${
CUDA_SHORT
}
\"
-Wait -NoNewWindow"
echo
"Done!"
rm
-f
"
${
CUDA_FILE
}
"
.github/workflows/cuda/cu102-Linux-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/usr/local/cuda-10.2
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5"
.github/workflows/cuda/cu102-Linux.sh
0 → 100755
View file @
14398229
#!/bin/bash
OS
=
ubuntu1804
wget
-nv
https://developer.download.nvidia.com/compute/cuda/repos/
${
OS
}
/x86_64/cuda-
${
OS
}
.pin
sudo mv
cuda-
${
OS
}
.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
-nv
https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-
${
OS
}
-10-2-local-10
.2.89-440.33.01_1.0-1_amd64.deb
sudo
dpkg
-i
cuda-repo-
${
OS
}
-10-2-local-10
.2.89-440.33.01_1.0-1_amd64.deb
sudo
apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo
apt-get
-qq
update
sudo
apt
install
-y
cuda-nvcc-10-2 cuda-libraries-dev-10-2
sudo
apt clean
rm
-f
https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-
${
OS
}
-10-2-local-10
.2.89-440.33.01_1.0-1_amd64.deb
.github/workflows/cuda/cu102-Windows-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/c/Program
\
Files/NVIDIA
\
GPU
\
Computing
\
Toolkit/CUDA/v10.2
PATH
=
${
CUDA_HOME
}
/bin:
$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5"
.github/workflows/cuda/cu102-Windows.sh
0 → 100755
View file @
14398229
#!/bin/bash
# Install NVIDIA drivers, see:
# https://github.com/pytorch/vision/blob/master/packaging/windows/internal/cuda_install.bat#L99-L102
curl
-k
-L
"https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download"
--output
"/tmp/gpu_driver_dlls.zip"
7z x
"/tmp/gpu_driver_dlls.zip"
-o
"/c/Windows/System32"
export
CUDA_SHORT
=
10.2
export
CUDA_URL
=
https://developer.download.nvidia.com/compute/cuda/
${
CUDA_SHORT
}
/Prod/local_installers
export
CUDA_FILE
=
cuda_
${
CUDA_SHORT
}
.89_441.22_win10.exe
# Install CUDA:
curl
-k
-L
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
--output
"
${
CUDA_FILE
}
"
echo
""
echo
"Installing from
${
CUDA_FILE
}
..."
PowerShell
-Command
"Start-Process -FilePath
\"
${
CUDA_FILE
}
\"
-ArgumentList
\"
-s nvcc_
${
CUDA_SHORT
}
cuobjdump_
${
CUDA_SHORT
}
nvprune_
${
CUDA_SHORT
}
cupti_
${
CUDA_SHORT
}
cublas_dev_
${
CUDA_SHORT
}
cudart_
${
CUDA_SHORT
}
cufft_dev_
${
CUDA_SHORT
}
curand_dev_
${
CUDA_SHORT
}
cusolver_dev_
${
CUDA_SHORT
}
cusparse_dev_
${
CUDA_SHORT
}
npp_dev_
${
CUDA_SHORT
}
nvrtc_dev_
${
CUDA_SHORT
}
nvml_dev_
${
CUDA_SHORT
}
\"
-Wait -NoNewWindow"
echo
"Done!"
rm
-f
"
${
CUDA_FILE
}
"
.github/workflows/cuda/cu111-Linux-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/usr/local/cuda-11.1
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
.github/workflows/cuda/cu111-Linux.sh
0 → 100755
View file @
14398229
#!/bin/bash
OS
=
ubuntu1804
wget
-nv
https://developer.download.nvidia.com/compute/cuda/repos/
${
OS
}
/x86_64/cuda-
${
OS
}
.pin
sudo mv
cuda-
${
OS
}
.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
-nv
https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda-repo-
${
OS
}
-11-1-local_11
.1.1-455.32.00-1_amd64.deb
sudo
dpkg
-i
cuda-repo-
${
OS
}
-11-1-local_11
.1.1-455.32.00-1_amd64.deb
sudo
apt-key add /var/cuda-repo-
${
OS
}
-11-1-local
/7fa2af80.pub
sudo
apt-get
-qq
update
sudo
apt
install
-y
cuda-nvcc-11-1 cuda-libraries-dev-11-1
sudo
apt clean
rm
-f
https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda-repo-
${
OS
}
-11-1-local_11
.1.1-455.32.00-1_amd64.deb
.github/workflows/cuda/cu111-Windows-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/c/Program
\
Files/NVIDIA
\
GPU
\
Computing
\
Toolkit/CUDA/v11.1
PATH
=
${
CUDA_HOME
}
/bin:
$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"6.0+PTX"
.github/workflows/cuda/cu111-Windows.sh
0 → 100755
View file @
14398229
#!/bin/bash
# Install NVIDIA drivers, see:
# https://github.com/pytorch/vision/blob/master/packaging/windows/internal/cuda_install.bat#L99-L102
curl
-k
-L
"https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download"
--output
"/tmp/gpu_driver_dlls.zip"
7z x
"/tmp/gpu_driver_dlls.zip"
-o
"/c/Windows/System32"
export
CUDA_SHORT
=
11.1
export
CUDA_URL
=
https://developer.download.nvidia.com/compute/cuda/
${
CUDA_SHORT
}
.1/local_installers
export
CUDA_FILE
=
cuda_
${
CUDA_SHORT
}
.1_456.81_win10.exe
# Install CUDA:
curl
-k
-L
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
--output
"
${
CUDA_FILE
}
"
echo
""
echo
"Installing from
${
CUDA_FILE
}
..."
PowerShell
-Command
"Start-Process -FilePath
\"
${
CUDA_FILE
}
\"
-ArgumentList
\"
-s nvcc_
${
CUDA_SHORT
}
cuobjdump_
${
CUDA_SHORT
}
nvprune_
${
CUDA_SHORT
}
cupti_
${
CUDA_SHORT
}
cublas_dev_
${
CUDA_SHORT
}
cudart_
${
CUDA_SHORT
}
cufft_dev_
${
CUDA_SHORT
}
curand_dev_
${
CUDA_SHORT
}
cusolver_dev_
${
CUDA_SHORT
}
cusparse_dev_
${
CUDA_SHORT
}
npp_dev_
${
CUDA_SHORT
}
nvrtc_dev_
${
CUDA_SHORT
}
nvml_dev_
${
CUDA_SHORT
}
\"
-Wait -NoNewWindow"
echo
"Done!"
rm
-f
"
${
CUDA_FILE
}
"
.github/workflows/cuda/cu113-Linux-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/usr/local/cuda-11.3
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
.github/workflows/cuda/cu113-Linux.sh
0 → 100755
View file @
14398229
#!/bin/bash
OS
=
ubuntu1804
wget
-nv
https://developer.download.nvidia.com/compute/cuda/repos/
${
OS
}
/x86_64/cuda-
${
OS
}
.pin
sudo mv
cuda-
${
OS
}
.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
-nv
https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-
${
OS
}
-11-3-local_11
.3.0-465.19.01-1_amd64.deb
sudo
dpkg
-i
cuda-repo-
${
OS
}
-11-3-local_11
.3.0-465.19.01-1_amd64.deb
sudo
apt-key add /var/cuda-repo-
${
OS
}
-11-3-local
/7fa2af80.pub
sudo
apt-get
-qq
update
sudo
apt
install
-y
cuda-nvcc-11-3 cuda-libraries-dev-11-3
sudo
apt clean
rm
-f
https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-
${
OS
}
-11-3-local_11
.3.0-465.19.01-1_amd64.deb
.github/workflows/cuda/cu113-Windows-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/c/Program
\
Files/NVIDIA
\
GPU
\
Computing
\
Toolkit/CUDA/v11.3
PATH
=
${
CUDA_HOME
}
/bin:
$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"6.0+PTX"
.github/workflows/cuda/cu113-Windows.sh
0 → 100755
View file @
14398229
#!/bin/bash
# Install NVIDIA drivers, see:
# https://github.com/pytorch/vision/blob/master/packaging/windows/internal/cuda_install.bat#L99-L102
curl
-k
-L
"https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download"
--output
"/tmp/gpu_driver_dlls.zip"
7z x
"/tmp/gpu_driver_dlls.zip"
-o
"/c/Windows/System32"
export
CUDA_SHORT
=
11.3
export
CUDA_URL
=
https://developer.download.nvidia.com/compute/cuda/
${
CUDA_SHORT
}
.0/local_installers
export
CUDA_FILE
=
cuda_
${
CUDA_SHORT
}
.0_465.89_win10.exe
# Install CUDA:
curl
-k
-L
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
--output
"
${
CUDA_FILE
}
"
echo
""
echo
"Installing from
${
CUDA_FILE
}
..."
PowerShell
-Command
"Start-Process -FilePath
\"
${
CUDA_FILE
}
\"
-ArgumentList
\"
-s nvcc_
${
CUDA_SHORT
}
cuobjdump_
${
CUDA_SHORT
}
nvprune_
${
CUDA_SHORT
}
cupti_
${
CUDA_SHORT
}
cublas_dev_
${
CUDA_SHORT
}
cudart_
${
CUDA_SHORT
}
cufft_dev_
${
CUDA_SHORT
}
curand_dev_
${
CUDA_SHORT
}
cusolver_dev_
${
CUDA_SHORT
}
cusparse_dev_
${
CUDA_SHORT
}
thrust_
${
CUDA_SHORT
}
npp_dev_
${
CUDA_SHORT
}
nvrtc_dev_
${
CUDA_SHORT
}
nvml_dev_
${
CUDA_SHORT
}
\"
-Wait -NoNewWindow"
echo
"Done!"
rm
-f
"
${
CUDA_FILE
}
"
.github/workflows/cuda/cu115-Linux-env.sh
0 → 100644
View file @
14398229
#!/bin/bash
CUDA_HOME
=
/usr/local/cuda-11.5
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
.github/workflows/cuda/cu115-Linux.sh
0 → 100755
View file @
14398229
#!/bin/bash
OS
=
ubuntu1804
wget
-nv
https://developer.download.nvidia.com/compute/cuda/repos/
${
OS
}
/x86_64/cuda-
${
OS
}
.pin
sudo mv
cuda-
${
OS
}
.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
-nv
https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda-repo-
${
OS
}
-11-5-local_11
.5.2-495.29.05-1_amd64.deb
sudo
dpkg
-i
cuda-repo-
${
OS
}
-11-5-local_11
.5.2-495.29.05-1_amd64.deb
sudo
apt-key add /var/cuda-repo-
${
OS
}
-11-5-local
/7fa2af80.pub
sudo
apt-get
-qq
update
sudo
apt
install
-y
cuda-nvcc-11-5 cuda-libraries-dev-11-5
sudo
apt clean
rm
-f
https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda-repo-
${
OS
}
-11-5-local_11
.5.2-495.29.05-1_amd64.deb
Prev
1
2
3
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment