Skip to content

Commit

Permalink
pip install instead of conda for torch nightly on linux
Browse files Browse the repository at this point in the history
  • Loading branch information
mcmonkey4eva committed Aug 14, 2024
1 parent cdf06ef commit 45452fb
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,8 @@ runs:
shell: bash -el {0}
run: |
conda activate gha-comfyui-${{ inputs.python_version }}-${{ inputs.torch_version }}
conda install pytorch torchvision torchaudio pytorch-cuda=${{ inputs.cuda_version }} -c pytorch-nightly -c nvidia
#conda install pytorch torchvision torchaudio pytorch-cuda=${{ inputs.cuda_version }} -c pytorch-nightly -c nvidia
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
- name: '[Unix-Linux-Only] Install Pytorch stable'
if: ${{ inputs.os == 'linux' && inputs.torch_version == 'stable' }}
Expand Down

0 comments on commit 45452fb

Please sign in to comment.