rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

updates PyTorch to allow AMD gpu splitting

Browse Source 
Feature adding update for AMD GPUs:
Changes the installed PyTorch version from 1.12.1+rocm5.1.1 to the newer stable version 1.13.1+rocm5.2 to allow AMD GPUs to utilize VRAM splitting across multiple GPUs
This commit is contained in:
YellowRoseCx
2023-02-20 18:40:21 -06:00
committed by GitHub
parent b3083034ea
commit 83bd1f0db6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
environments/rocm.yml
View File

@@ -22,8 +22,8 @@ dependencies:
- Pillow - Pillow
- psutil - psutil
- pip: - pip:
- --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 - --extra-index-url https://download.pytorch.org/whl/rocm5.2
- torch==1.12.1+rocm5.1.1 - torch==1.13.1+rocm5.2
- flask-cloudflared==0.0.10 - flask-cloudflared==0.0.10
- flask-ngrok - flask-ngrok
- flask-cors - flask-cors