diff --git a/.jenkins/pytorch/perf_test/test_cpu_speed_mnist.sh b/.jenkins/pytorch/perf_test/test_cpu_speed_mnist.sh index 3048b9f20be8..29086fbc9976 100644 --- a/.jenkins/pytorch/perf_test/test_cpu_speed_mnist.sh +++ b/.jenkins/pytorch/perf_test/test_cpu_speed_mnist.sh @@ -12,7 +12,7 @@ test_cpu_speed_mnist () { cd examples/mnist - pip install -q -r requirements.txt + conda install -c pytorch torchvision-cpu # Download data python main.py --epochs 0 diff --git a/.jenkins/pytorch/perf_test/test_gpu_speed_mnist.sh b/.jenkins/pytorch/perf_test/test_gpu_speed_mnist.sh index 84685419542a..2453f3d70e70 100644 --- a/.jenkins/pytorch/perf_test/test_gpu_speed_mnist.sh +++ b/.jenkins/pytorch/perf_test/test_gpu_speed_mnist.sh @@ -12,7 +12,7 @@ test_gpu_speed_mnist () { cd examples/mnist - pip install -q -r requirements.txt + conda install -c pytorch torchvision # Download data python main.py --epochs 0