Skip to content

Commit e70ed0c

Browse files
committed
using miniconda venv instead of OS venv ; increasing the memory requirement for using ramdisks
1 parent 1a7e65c commit e70ed0c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

gpu/install_gpu_driver.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ function is_ramdisk() {
325325
function mount_ramdisk(){
326326
local free_mem
327327
free_mem="$(awk '/^MemFree/ {print $2}' /proc/meminfo)"
328-
if [[ ${free_mem} -lt 10500000 ]]; then return 0 ; fi
328+
if [[ ${free_mem} -lt 20500000 ]]; then return 0 ; fi
329329

330330
# Write to a ramdisk instead of churning the persistent disk
331331

@@ -1792,7 +1792,7 @@ function download_gpu_monitoring_agent(){
17921792

17931793
function install_gpu_monitoring_agent_dependency(){
17941794
cd /opt/google/compute-gpu-monitoring/linux
1795-
/usr/bin/python3 -m venv venv
1795+
/opt/conda/miniconda3/bin/python3 -m venv venv
17961796
(
17971797
source venv/bin/activate
17981798
pip install wheel
@@ -1823,7 +1823,7 @@ function install_gpu_agent() {
18231823
| sed -e 's/-u --format=/--format=/' \
18241824
| dd status=none of="${install_dir}/report_gpu_metrics.py"
18251825
local venv="${install_dir}/venv"
1826-
/usr/bin/python3 -m venv "${venv}"
1826+
/opt/conda/miniconda3/bin/python3 -m venv "${venv}"
18271827
(
18281828
source "${venv}/bin/activate"
18291829
python3 -m pip install --upgrade pip

0 commit comments

Comments
 (0)