Update the Conda installation command to match the latest from PyTorch

triton-inference-server · May 7, 2024 · d01c4ec · d01c4ec
1 parent 95128b2
commit d01c4ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/gen_pb_exec_env.sh b/tools/gen_pb_exec_env.sh
@@ -41,7 +41,7 @@ export PYTHONNOUSERSITE=True
 conda install -c conda-forge libstdcxx-ng=12 -y
 
 # install PyTorch
-conda install pytorch torchvision torchaudio pytorch-cuda -c pytorch -c nvidia -y
+conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y
 
 # pack environment
 rm -f pb_exec_env_model.py.tar.gz