Update float_install.sh

Updating FLOAT install to allow ssh access to localhost
AFKD98 · Mar 3, 2024 · 0557c89 · 0557c89
1 parent d9f915a
commit 0557c89
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/float_install.sh b/float_install.sh
@@ -68,7 +68,8 @@ else
 
 fi
 
-
+#Add authorized key of localhost 
+cat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys
 
 if [ "$1" == "--cuda" ]; then
   wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
@@ -84,4 +85,4 @@ fi
 
 
 # Download the FEMNIST dataset
-./benchmark/dataset/download.sh download femnist
+./benchmark/dataset/download.sh download femnist