如何在Google Colab中持久化安装nsys避免重启后丢失
解决Google Colab重启后nsys消失的问题
Colab的虚拟机每次重启都会重置为初始状态,手动安装的工具自然会丢失。要实现重启后无需重新安装nsys,可用以下几种方案:
方案1:把安装命令做成自动执行的初始化代码块
把安装、配置nsys的完整命令放在笔记本顶部的代码单元格里,设置成启动时自动运行:
- 新建一个代码单元格,写入你的安装流程(以下是示例,替换成你实际用的版本和命令):
# 下载对应版本的nsight-system安装包 wget https://developer.download.nvidia.com/devtools/nsight-systems/2024.2.1/nsight-systems-2024.2.1_2024.2.1.11_linux.tar.gz # 解压到当前目录 tar -xzf nsight-systems-2024.2.1_2024.2.1.11_linux.tar.gz # 将nsys的bin目录加入环境变量 export PATH=$PATH:/content/nsight-systems-2024.2.1/bin
- 右键点击这个单元格,选择「Run before all other cells」,这样每次打开或重启笔记本时,这个单元格会自动执行,完成nsys的部署。
方案2:挂载Google Drive保存安装文件
把解压好的nsight-systems文件夹传到你的Google Drive,重启时直接挂载Drive并配置环境变量,省去重复下载解压的步骤:
- 先把解压后的nsight-systems文件夹上传到Drive的某个路径(比如
/MyDrive/Tools/nsight-systems/) - 在笔记本里添加挂载和配置的代码:
from google.colab import drive drive.mount('/content/drive') # 配置环境变量 !export PATH=$PATH:/content/drive/MyDrive/Tools/nsight-systems/bin
这种方法速度更快,适合频繁使用的场景。
方案3:用%store命令持久化环境变量(仅辅助)
如果已经完成安装,可先保存环境变量,重启后恢复:
# 保存当前PATH %store PATH
重启后执行:
# 恢复PATH %store -r PATH
注意:这个方法只能保存环境变量,不能保留已安装的文件,必须配合前两种方案使用。
内容的提问来源于stack exchange,提问作者CisMine
相关产品推荐
相关产品推荐

