theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed torch version; CUDA acceleration utils

Browse Source
This commit is contained in:
josc146
2023-05-23 11:19:39 +08:00
parent ecb5d6c6e4
commit 7989e93afe
6 changed files with 745 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend-golang/rwkv.go
View File

@@ -48,7 +48,7 @@ func (a *App) InstallPyDep(cnMirror bool) (string, error) {
return "", err
}
ChangeFileLine("./py310/python310._pth", 3, "Lib\\site-packages")
_, err = Cmd(python, "-m", "pip", "install", "torch", "torchvision", "torchaudio", "--index-url", "https://download.pytorch.org/whl/cu117")
_, err = Cmd(python, "-m", "pip", "install", "torch==1.13.1", "torchvision==0.14.1", "torchaudio==0.13.1", "--index-url", "https://download.pytorch.org/whl/cu117")
if err != nil {
return "", err
}