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 31
31
32
32
printf " Installing PyTorch with %s\n" " ${cudatoolkit} "
33
33
if [ " ${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} "
35
35
else
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} "
37
37
fi
38
38
39
39
printf " * Installing torchvision\n"
Original file line number Diff line number Diff line change 28
28
fi
29
29
30
30
printf " 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} "
32
32
33
33
torch_cuda=$( python -c " import torch; print(torch.cuda.is_available())" )
34
34
echo torch.cuda.is_available is $torch_cuda
You can’t perform that action at this time.
0 commit comments