Skip to content

Solution: Better Fix for Cuda Home on Linux #107

@RandomInternetPreson

Description

@RandomInternetPreson

Hello, I really want you to know how much I love this extension!!! OMG <3 Thank you so much, it is really appreciated.

The integration of deepspeed in particular lets one pretty much talk to their models in real time. I had been using your extension in windows using the prebuilt wheels and they were working great.

I recently switched over to ubuntu and got deepspeed working as per your instructions. The initialization of deepspeed is a little cumbersome and I had a lot of stuff in my CMD_FLAGS.txt file that I wanted to automatically run when starting oobabooga.

I figured out that putting:

export CUDA_HOME="/usr/local/cuda"

Right above:

setup installer env

in the start_linux.sh file is a lot easier and let's me use the CMD_Flags.txt file once again.

I have included an image to help make it more clear:

cudahome

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions