diff --git a/LabsSolutions/03-pytorch-gan/submit.py b/LabsSolutions/03-pytorch-gan/submit.py
index 105e913..c80bdbb 100755
--- a/LabsSolutions/03-pytorch-gan/submit.py
+++ b/LabsSolutions/03-pytorch-gan/submit.py
@@ -31,9 +31,8 @@ def makejob(dataset, nruns):
 cd $TMPDIR/tpGAN
 
 echo "Setting up the virtual environment"
-python3 -m pip install virtualenv --user
-virtualenv -p python3 venv
-source venv/bin/activate
+python3 -m venv $TMPDIR/venv
+source $TMPDIR/venv/bin/activate
 python -m pip install -r requirements.txt
 
 echo "Training"