[046]Google Colab 运行 ICLight 一键打光

Google Colab 提供了学习用的免费GPT服务器，本篇内容讲解如何一键在 Colab 上部署IC-Light 并通过 WEBUI 进行使用。

视频讲解：

本文章内容的视频版本：

https://youtu.be/RXsZrCSj0oA

想省事的，不想复制命令的直接点这个链接

https://colab.research.google.com/drive/1RDxmqrTBg2QZeIRtxRW6h-teQRn50hN2?usp=sharing

检测 CUDA：

 python
import torch  

# 打印PyTorch的版本
print("PyTorch版本:", torch.__version__)  

# 打印CUDA的版本
print("CUDA版本:", torch.version.cuda)  

# 检查CUDA是否可用
cuda_available = torch.cuda.is_available()
print("CUDA是否可用:", cuda_available)  

# 如果CUDA可用，打印当前CUDA设备的信息
if cuda_available:
    print("当前CUDA设备名称:", torch.cuda.get_device_name(0))
    print("当前CUDA设备总计算能力 (Capability):", torch.cuda.get_device_capability(0))

装载Google Drive

python
from google.colab import drive
drive.mount('/content/drive')
!apt install git
!git clone https://github.com/NiuGee/iclight-patch.git /content/drive/MyDrive
%cd /content/drive/MyDrive/IC-Light/

创建并使用虚拟环境

python
%cd /content/drive/MyDrive/IC-Light/
%pwd
!pip3 install virtualenv
!virtualenv venv
!source venv/bin/activate

安装pytorch

python
!pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121

安装其他依赖

python
%cd /content/drive/MyDrive/IC-Light/
!pip install -r requirements.txt

修改代码，添加gradio代理访问

runui添加share=True

运行代码：

!pwd
!python iclight-webui-niugee.com.py

访问界面

通过上文的第二个gradio映射的地址即可以访问到工程界面了

注意事项

每个账户 有免费额度的，用没了就没了。。。。

云盘根目录是 “/content/drive/My Drive”，这里有个空格，在输入命令的时候空格要加转义\。比如：

!python /content/drive/My\ Drive/BertNer/BERT_NER.py

import 时出现如果出现类似下面的报错：

ModuleNotFoundError: No module named ‘XXXX’

解决方案：将项目根目录添加到环境变量中:

import sys
cur = '/content/drive/My Drive/XXXX'
sys.path.append(cur)

聚沙成塔 汇水成渊

目前所有的AI项目只能本地，或者蹭免费云，牛哥正在发起 GPU 服务器众筹项目。
有心的支持一下：

https://niugee.com/make_a_love/?source=046