windows安装 cuda

前情回顾

因为要测试bark生成语音,发现CPU运算的速度太慢了,得使用GPU,目前笔记本上的1050T Nvida还能用,所以要测试一下GPU版本的bark代码,故需要安装gpu的torch。

安装CPU版本

pip install torch

安装GPU版本

torch_stable 下载对应的GPU版本,手动安装

pip install .\torch-2.0.1+cu118-cp310-cp310-win_amd64.whl

再安装其他软件

pip install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

参考

使用GPU版本的torch


作者:spike

分类: Python

创作时间:2023-08-12

更新时间:2024-12-09

联系方式放在中括号之中例如[[email protected]],回复评论在开头加上标号例如:#1