Skip to content

Commit

Permalink
loosen cudnn
Browse files Browse the repository at this point in the history
  • Loading branch information
hmaarrfk authored Sep 21, 2024
1 parent 66c1a1d commit 6faa699
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions recipe/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ outputs:
- libcusparse-dev # [build_platform != target_platform]
{% endif %}
# optional dependencies for CUDA 11.2+
- cudnn ~=8.8
- cudnn
- nccl ~=2.16 # [build_platform != target_platform]
- cutensor ~=2.0 # [build_platform != target_platform]
host:
Expand All @@ -181,7 +181,7 @@ outputs:
- libcusparse-dev
{% endif %}
# optional dependencies
- cudnn ~=8.8
- cudnn
- nccl ~=2.16 # [linux]
- cutensor ~=2.0
run:
Expand Down

0 comments on commit 6faa699

Please sign in to comment.