52 lines
1.9 KiB
YAML
52 lines
1.9 KiB
YAML
|
on:
|
||
|
|
push:
|
||
|
|
jobs:
|
||
|
|
linux-x86_64-cuda-11-2:
|
||
|
|
runs-on: ubuntu-18.04
|
||
|
|
steps:
|
||
|
|
- name: Cancel Previous Runs
|
||
|
|
uses: styfle/cancel-workflow-action@0.8.0
|
||
|
|
with:
|
||
|
|
access_token: ${{ github.token }}
|
||
|
|
- name: Maximize build space
|
||
|
|
uses: easimon/maximize-build-space@master
|
||
|
|
with:
|
||
|
|
root-reserve-mb: 512
|
||
|
|
swap-size-mb: 8192
|
||
|
|
remove-dotnet: 'true'
|
||
|
|
remove-haskell: 'true'
|
||
|
|
- uses: actions/checkout@v2
|
||
|
|
- uses: konduitai/cuda-install/.github/actions/install-cuda-ubuntu@master
|
||
|
|
env:
|
||
|
|
cuda: 11.2.1_461
|
||
|
|
GCC: 9
|
||
|
|
- uses: ./.github/actions/install-protobuf-linux
|
||
|
|
- name: Set up Java for publishing to GitHub Packages
|
||
|
|
uses: actions/setup-java@v1
|
||
|
|
with:
|
||
|
|
java-version: 1.8
|
||
|
|
server-id: sonatype-nexus-snapshots
|
||
|
|
server-username: MAVEN_USERNAME
|
||
|
|
server-password: MAVEN_PASSWORD
|
||
|
|
gpg-private-key: ${{ secrets.SONATYPE_GPG_KEY }}
|
||
|
|
gpg-passphrase: MAVEN_GPG_PASSPHRASE
|
||
|
|
- name: Run cuda compilation on linux-x86_64
|
||
|
|
shell: bash
|
||
|
|
env:
|
||
|
|
DEBIAN_FRONTEND: noninteractive
|
||
|
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||
|
|
PUBLISH_TO: ossrh
|
||
|
|
MAVEN_USERNAME: ${{ secrets.SONATYPE_USER_1 }}
|
||
|
|
MAVEN_PASSWORD: ${{ secrets.SONATYPE_USER1_PASS }}
|
||
|
|
MAVEN_GPG_PASSPHRASE: ${{ secrets.PACKAGES_GPG_PASS }}
|
||
|
|
run: |
|
||
|
|
export PATH="/usr/local/cuda-11.2/bin:$PATH"
|
||
|
|
nvcc --version
|
||
|
|
mvn --version
|
||
|
|
cmake --version
|
||
|
|
protoc --version
|
||
|
|
sudo apt-get autoremove
|
||
|
|
sudo apt-get clean
|
||
|
|
bash ./change-cuda-versions.sh 11.2
|
||
|
|
mvn -Possrh -Djavacpp.platform=linux-x86_64 -Dlibnd4j.compute="5.0 5.2 5.3 6.0 6.2 8.0" -pl ":nd4j-cuda-11.2,:deeplearning4j-cuda-11.2,:libnd4j" --also-make -Dlibnd4j.chip=cuda --batch-mode deploy -DskipTests
|