pull/332/head
Akegarasu 2023-12-24 20:10:07 +08:00
commit a586c8a3f1
No known key found for this signature in database
GPG Key ID: 9D600258808ACBCD
1 changed files with 1 additions and 1 deletions

View File

@ -30,7 +30,7 @@ cuda_minor_version=$(echo "$cuda_version" | awk -F'.' '{print $2}')
echo "Cuda Version:$cuda_version"
if (( cuda_major_version >= 12 )) then
if (( cuda_major_version >= 12 )); then
echo "install torch 2.1.0+cu121"
pip install torch==2.1.2+cu121 torchvision==0.16.2+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
pip install --no-deps xformers===0.0.23.post1