Ubuntu

Luo-Littile

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

在Ubuntu下跑，软件是直接克隆开源的，可以正常运行，但是直接就在CPU上运行了，要用GPU 怎么整

烟鬼

不懂帮顶

gaolongcc

tf的版本太高了，重新用conda创建一个环境
conda create -n deepfacelab -c main python=3.7 cudnn=7.6.5 cudatoolkit=10.1.243
然后把requirements-cuda.txt里的tf版本改成2.3.1
pip install -r requirements-cuda.txt  -i https://pypi.tuna.tsinghua.edu.cn/simple
就好了
加油加油

Luo-Littile

gaolongcc 发表于 2022-3-10 16:41
tf的版本太高了，重新用conda创建一个环境
conda create -n deepfacelab -c main python=3.7 cudnn=7.6.5 c ...

多谢多谢  我试试