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
gaoqiong
flash-attention
Commits
a682252b
Commit
a682252b
authored
Jun 02, 2023
by
Pierce Freeman
Browse files
OS version build numbers
parent
cd0c169e
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
24 additions
and
9 deletions
+24
-9
.github/workflows/cuda/cu102-Linux.sh
.github/workflows/cuda/cu102-Linux.sh
+3
-1
.github/workflows/cuda/cu113-Linux.sh
.github/workflows/cuda/cu113-Linux.sh
+7
-1
.github/workflows/cuda/cu116-Linux.sh
.github/workflows/cuda/cu116-Linux.sh
+3
-1
.github/workflows/publish.yml
.github/workflows/publish.yml
+11
-6
No files found.
.github/workflows/cuda/cu102-Linux.sh
View file @
a682252b
#!/bin/bash
OS
=
ubuntu1804
# Strip the periods from the version number
OS_VERSION
=
$(
echo
$OS_VERSION
|
tr
-d
.
)
OS
=
ubuntu
${
OS_VERSION
}
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
...
...
.github/workflows/cuda/cu113-Linux.sh
View file @
a682252b
#!/bin/bash
OS
=
ubuntu1804
# Strip the periods from the version number
OS_VERSION
=
$(
echo
$OS_VERSION
|
tr
-d
.
)
OS
=
ubuntu
${
OS_VERSION
}
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
# TODO: If on version < 22.04, install via signal-desktop-keyring
# For future versions it's deprecated and should be moved into the trusted folder
# sudo mv /var/cuda-repo-${OS}-11-3-local/7fa2af80.pub /etc/apt/trusted.gpg.d/
sudo
apt-key add /var/cuda-repo-
${
OS
}
-11-3-local
/7fa2af80.pub
sudo
apt-get
-qq
update
...
...
.github/workflows/cuda/cu116-Linux.sh
View file @
a682252b
#!/bin/bash
OS
=
ubuntu1804
# Strip the periods from the version number
OS_VERSION
=
$(
echo
$OS_VERSION
|
tr
-d
.
)
OS
=
ubuntu
${
OS_VERSION
}
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
...
...
.github/workflows/publish.yml
View file @
a682252b
...
...
@@ -37,12 +37,15 @@ jobs:
fail-fast
:
false
matrix
:
os
:
[
ubuntu-20.04
,
ubuntu-22.04
]
python-version
:
[
'
3.7'
,
'
3.8'
,
'
3.9'
,
'
3.10'
]
torch-version
:
[
1.11.0
,
1.12.0
,
1.12.1
]
cuda-version
:
[
'
113'
,
'
116'
]
exclude
:
-
torch-version
:
1.11.0
cuda-version
:
'
116'
#python-version: ['3.7', '3.8', '3.9', '3.10']
python-version
:
[
'
3.10'
]
torch-version
:
[
1.11.0
]
cuda-version
:
[
'
113'
]
#torch-version: [1.11.0, 1.12.0, 1.12.1]
#cuda-version: ['113', '116']
#exclude:
# - torch-version: 1.11.0
# cuda-version: '116'
steps
:
-
name
:
Checkout
...
...
@@ -65,6 +68,8 @@ jobs:
-
name
:
Install CUDA ${{ matrix.cuda-version }}
if
:
${{ matrix.cuda-version != 'cpu' }}
env
:
OS_VERSION
:
${{ runner.release }}
run
:
|
bash .github/workflows/cuda/cu${{ matrix.cuda-version }}-${{ runner.os }}.sh
shell
:
...
...
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