Skip to content

Commit

Permalink
Added comments with todo to remove fix
Browse files Browse the repository at this point in the history
  • Loading branch information
LujieDuan committed Nov 5, 2024
1 parent c18d268 commit 14e0263
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ setup_repo() {
install_cuda_from_package_manager() {
setup_repo
install_driver_package
# TODO(b/377558109): remove the temporary fix once the repo is updated
sudo yum -y install cuda-toolkit cuda-demo*
verify_driver
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ case $DEVICE_CODE in
;;
*)
# For newer GPUs, install the latest version
# TODO(b/377558109): remove the temporary fix once the repo is updated
sudo apt -y install cuda-12-6
;;
esac
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ setup_repo() {
install_cuda_from_package_manager() {
setup_repo
install_driver_package
# TODO(b/377558109): remove the temporary fix once the repo is updated
sudo yum -y install cuda-toolkit cuda-demo*
verify_driver
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ case $DEVICE_CODE in
;;
*)
# For newer GPUs, install the latest version
# TODO(b/377558109): remove the temporary fix once the repo is updated
sudo apt -y install cuda-12-6
;;
esac
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ setup_repo() {
install_cuda_from_package_manager() {
setup_repo
install_driver_package
# TODO(b/377558109): remove the temporary fix once the repo is updated
sudo yum -y install cuda-toolkit cuda-demo*
verify_driver
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ case $DEVICE_CODE in
;;
*)
# For newer GPUs, install the latest version
# TODO(b/377558109): remove the temporary fix once the repo is updated
sudo apt -y install cuda-12-6
;;
esac
Expand Down

0 comments on commit 14e0263

Please sign in to comment.