Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
kamo-naoyuki committed May 12, 2022
1 parent b0050d9 commit 8fbac77
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/installers/install_torch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ install_torch(){
log conda install -y "pytorch=${torch_version}" "torchaudio=$1" cpuonly -c pytorch
conda install -y "pytorch=${torch_version}" "torchaudio=$1" cpuonly -c pytorch
elif [ "${cuda_version}" = "11.5" ]; then
# NOTE(kamo): In my environment, conda-forge only could installed, but I don't know why @ 12, May, 2022
# NOTE(kamo): In my environment, cudatoolkit of conda-forge only could be installed, but I don't know why @ 12, May, 2022
cudatoolkit_channel=conda-forge
log conda install -y "pytorch=${torch_version}" "torchaudio=$1" "cudatoolkit=${cuda_version}" -c pytorch -c "${cudatoolkit_channel}"
conda install -y "pytorch=${torch_version}" "torchaudio=$1" "cudatoolkit=${cuda_version}" -c pytorch -c "${cudatoolkit_channel}"
Expand Down

0 comments on commit 8fbac77

Please sign in to comment.