OpenSora初体验

一、简介

打开

GitHub - hpcaitech/Open-Sora: Open-Sora: Democratizing Efficient Video Production for All

你可以发现3月18日Open-Sora 1.0 发布的消息：

快速跳到安装部分：

二、准备体验机器

打开

InternStudio​studio.intern-ai.org.cn/

新建一个InternStudio的机器：sora-test

进入开发机：

nvidia-smi

nvcc -V

注意：cuda版本是11.7

三、准备CUDA 11.7的conda环境

1、创建conda环境

conda create -n opensora python=3.10 -y conda activate opensora

2、安装Pytorch

打开 https://pytorch.org/get-started/locally/

这里面只有CUDA11.8的。有两种方式：

1是用conda安装cuda 12.1或者cuda 11.8的版本，命令为：

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

2是选择历史版本，

安装方法为：

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

或

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2

为了省事点，使用最后一种方式吧！

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2

3、查看torch安装结果

python

import torch

torch.cuda.is_available()

torch.__version__ quit()

4、装flash-attn (optional)

pip install packaging ninja

已经装了。

pip install flash-attn --no-build-isolation

好像卡住了。先断掉。这是可选的。要不然先忽略掉。

5、安装apex (optional)

pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex.git