File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -37,13 +37,15 @@ conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
3737conda create -n pt python=3.12 -y
3838conda activate pt
3939conda install -c conda-forge conda-pack -y
40+ conda install pip
4041
4142# pre install step
4243export PYTHONNOUSERSITE=True
4344conda install -c conda-forge libstdcxx-ng=15 -y
4445
45- # install PyTorch
46- conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -y
46+ # install PyTorch (torch from pip to avoid conda version issues)
47+ pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130
48+ conda install torchaudio pytorch-cuda=13.0 -c pytorch -c nvidia -y
4749
4850# pack environment
4951rm -f pb_exec_env_model.py.tar.gz
You can’t perform that action at this time.
0 commit comments