File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 3131
3232printf " Installing PyTorch with %s\n" " ${cudatoolkit} "
3333if [ " ${os} " == " MacOSX" ]; then
34- conda install -y -c " pytorch-${UPLOAD_CHANNEL} " " pytorch-${UPLOAD_CHANNEL} " ::pytorch " ${cudatoolkit} " pytest
34+ conda install -y -c " pytorch-${UPLOAD_CHANNEL} " " pytorch-${UPLOAD_CHANNEL} " ::pytorch " ${cudatoolkit} "
3535else
36- conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia " pytorch-${UPLOAD_CHANNEL} " ::pytorch[build= " *${version} *" ] " ${cudatoolkit} " pytest
36+ conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia " pytorch-${UPLOAD_CHANNEL} " ::pytorch[build= " *${version} *" ] " ${cudatoolkit} "
3737fi
3838
3939printf " * Installing torchvision\n"
Original file line number Diff line number Diff line change 2828fi
2929
3030printf " Installing PyTorch with %s\n" " ${cudatoolkit} "
31- conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia " pytorch-${UPLOAD_CHANNEL} " ::pytorch[build= " *${version} *" ] " ${cudatoolkit} " pytest
31+ conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia " pytorch-${UPLOAD_CHANNEL} " ::pytorch[build= " *${version} *" ] " ${cudatoolkit} "
3232
3333torch_cuda=$( python -c " import torch; print(torch.cuda.is_available())" )
3434echo torch.cuda.is_available is $torch_cuda
You can’t perform that action at this time.
0 commit comments