[BugFix]: Conda to pip for circleci (#310)

Author: vmoens

.circleci/unittest/linux/scripts/install.sh

@@ -36,9 +36,11 @@ printf "Installing PyTorch with %s\n" "${CU_VERSION}"
 if [ "${CU_VERSION:-}" == cpu ] ; then
     # conda install -y pytorch torchvision cpuonly -c pytorch-nightly
     # use pip to install pytorch as conda can frequently pick older release
-    conda install -y pytorch torchvision cpuonly -c pytorch-nightly
+#    conda install -y pytorch torchvision cpuonly -c pytorch-nightly
+    pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 else
-    conda install -y pytorch torchvision cudatoolkit=11.3 -c pytorch-nightly
+#    conda install -y pytorch torchvision cudatoolkit=11.3 -c pytorch-nightly
+    pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu113
 fi
 
 printf "Installing functorch\n"

.circleci/unittest/linux/scripts/setup_env.sh

@@ -57,7 +57,7 @@ if [[ $OSTYPE == 'darwin'* ]]; then
   PRIVATE_MUJOCO_GL=glfw
 else
   conda install -y -c conda-forge mesa
-  conda install -y -c menpo osmesa
+  conda install -y -c conda-forge mesalib
   PRIVATE_MUJOCO_GL=osmesa
 fi
 

.circleci/unittest/linux_optdeps/scripts/install.sh

@@ -36,9 +36,11 @@ printf "Installing PyTorch with %s\n" "${CU_VERSION}"
 if [ "${CU_VERSION:-}" == cpu ] ; then
     # conda install -y pytorch torchvision cpuonly -c pytorch-nightly
     # use pip to install pytorch as conda can frequently pick older release
-    conda install -y pytorch cpuonly -c pytorch-nightly
+#    conda install -y pytorch cpuonly -c pytorch-nightly
+    pip3 install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 else
-    conda install -y pytorch cudatoolkit=11.3 -c pytorch-nightly
+#    conda install -y pytorch cudatoolkit=11.3 -c pytorch-nightly
+    pip3 install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cu113
 fi
 
 printf "Installing functorch\n"

.circleci/unittest/linux_stable/scripts/install.sh

@@ -37,9 +37,11 @@ printf "Installing PyTorch with %s\n" "${CU_VERSION}"
 if [ "${CU_VERSION:-}" == cpu ] ; then
     # conda install -y pytorch torchvision cpuonly -c pytorch-nightly
     # use pip to install pytorch as conda can frequently pick older release
-    conda install -y pytorch torchvision cpuonly -c pytorch
+#    conda install -y pytorch torchvision cpuonly -c pytorch
+    pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
 else
-    conda install -y pytorch torchvision cudatoolkit=11.3 -c pytorch
+#    conda install -y pytorch torchvision cudatoolkit=11.3 -c pytorch
+    pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
 fi
 
 printf "Installing functorch\n"

.circleci/unittest/linux_stable/scripts/setup_env.sh

@@ -58,7 +58,7 @@ if [[ $OSTYPE == 'darwin'* ]]; then
   PRIVATE_MUJOCO_GL=glfw
 else
   conda install -y -c conda-forge mesa
-  conda install -y -c menpo osmesa
+  conda install -y -c conda-forge mesalib
   PRIVATE_MUJOCO_GL=osmesa
 fi
 

test/test_transforms.py

@@ -219,8 +219,6 @@ def test_vecnorm_parallel_auto(nprc):
     for idx in range(nprc):
         queues[idx][1].put(msg)
 
-    for p in prcs:
-        p.join()
     obs_sum = td.get("next_observation_sum").clone()
     obs_ssq = td.get("next_observation_ssq").clone()
     obs_count = td.get("next_observation_count").clone()
@@ -258,7 +256,10 @@ def test_vecnorm_parallel_auto(nprc):
     msg = "all_done"
     for idx in range(nprc):
         queues[idx][1].put(msg)
+
     del queues
+    for p in prcs:
+        p.join()
 
 
 def _run_parallelenv(parallel_env, queue_in, queue_out):