Skip to content

Commit

Permalink
Remove cudnn_path
Browse files Browse the repository at this point in the history
  • Loading branch information
xhochy committed Sep 30, 2024
1 parent 937e296 commit 94d311d
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion recipe/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,6 @@ if [[ "${cuda_compiler_version:-None}" != "None" ]]; then

CUDA_ARGS="--enable_cuda \
--enable_nccl \
--cudnn_path=${PREFIX} \
--cuda_compute_capabilities=$TF_CUDA_COMPUTE_CAPABILITIES \
--cuda_version=$TF_CUDA_VERSION \
--cudnn_version=$TF_CUDNN_VERSION"
Expand Down

0 comments on commit 94d311d

Please sign in to comment.