[go: nahoru, domu]
Skip to content

Commit

Permalink
Add libcublas-12-2 to prevent the libnvinfer8 dependency from pulling…
Browse files Browse the repository at this point in the history 
… cuda 12.3 dependencies.

Otherwise, when pulled, the 12.3 will change the alias of /usr/local/cuda to 12.3, causing failures.

PiperOrigin-RevId: 575799576
  • Loading branch information
@MichaelHudgins @tensorflower-gardener
MichaelHudgins authored and tensorflower-gardener committed Oct 23, 2023
1 parent 6f30fc1 commit a02bae3
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ libcufft-12-2
libcurand-12-2
libcusolver-dev-12-2
libcusparse-dev-12-2
libcublas-12-2
libcublas-dev-12-2
libnccl-dev=2.18.5-1+cuda12.2
# CuDNN: https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#ubuntu-network-installation
Expand Down

0 comments on commit a02bae3

Please sign in to comment.