Add cuda-nvcc (#623)

* Add cuda nvcc to the requirements and build pipeline
- Enables system-specific optimizations
* Pin nvcc version and add channel to build command
* Bump pre-release version and remove manual build filter

.conda/meta.yaml

@@ -16,7 +16,7 @@ about:
  summary: {{ data.get('description') }}
 
 build:
- number: 3
+ number: 0
 
 source:
  path: ../
@@ -32,6 +32,7 @@ requirements:
  - python=3.7
  - cudatoolkit=11.3.1
  - cudnn=8.2.1
+ - nvidia::cuda-nvcc=11.3
  - sleap::tensorflow=2.7.0
  - sleap::pyside2=5.14.1
  - pillow

.github/workflows/build.yml

@@ -11,7 +11,6 @@ on:
  # branches:
  # - main
  # - develop
- # - talmo/120_fixes
 
 
 jobs:
@@ -73,14 +72,14 @@ jobs:
  # export DISPLAY=":1"
  # xvfb-run pytest tests/
  run: |
- conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/
+ conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/ -c https://conda.anaconda.org/nvidia/
  - name: Build conda package (Windows)
  if: matrix.os == 'windows-2019'
  shell: powershell
  run: |
  conda activate sleap
  pytest tests/
- conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/
+ conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/ -c https://conda.anaconda.org/nvidia/
 
  # Upload conda package
  - name: Login to Anaconda (Ubuntu)

.github/workflows/build_manual.yml

@@ -72,14 +72,14 @@ jobs:
  # export DISPLAY=":1"
  # xvfb-run pytest tests/
  run: |
- conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/
+ conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/ -c https://conda.anaconda.org/nvidia/
  - name: Build conda package (Windows)
  if: matrix.os == 'windows-2019'
  shell: powershell
  run: |
  conda activate sleap
  pytest tests/
- conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/
+ conda build .conda --output-folder build -c https://conda.anaconda.org/sleap/ -c https://conda.anaconda.org/nvidia/
 
  # Upload conda package
  - name: Login to Anaconda (Ubuntu)

environment.yml

@@ -4,6 +4,7 @@ dependencies:
 - python=3.7
 - cudatoolkit=11.3.1
 - cudnn=8.2.1
+- nvidia::cuda-nvcc=11.3
 # - sleap::tensorflow=2.7.0
 # - sleap::pyside2=5.14.1
 - pillow

sleap/version.py

@@ -12,7 +12,7 @@
 """
 
 
-__version__ = "1.2.0a1"
+__version__ = "1.2.0a2"
 
 
 def versions():