本文最后更新于 2025-02-25,文章内容可能已经过时。

在AutoDL上跑深度学习常用的命令

初始化虚拟环境

简单情况下使用

conda init

标准情况下使用

输入:vim ~/.bashrc
进入编辑页面:i
移动到最后一行加上:source /root/miniconda3/etc/profile.d/conda.sh
按ESC键,输入:wq
输入bash刷新终端

查看PyTorch版本和CUDA版本

import torch
print(torch.__version__)  # 输出PyTorch版本,例如:2.1.0
print(torch.version.cuda)  # 若安装了CUDA版本,显示对应的CUDA版本号

验证CUDA是否可用

import torch
print(torch.cuda。is_available())

查看虚拟环境

conda info --envs

解压文件

ZIP压缩包

unzip 文件名.zip

tar.gz压缩包

tar -zxvf 文件名.tar.gz

压缩文件夹

tar -cvf 文件夹名.tar.gz

zip -r 文件名.zip 文件夹名

移动文件夹

mv A B 
#将A文件夹及其包含的内容移动到B文件夹下

复制文件夹

cp -r A/B/
#将A文件夹以及包含的内容复制到B文件夹下

训练完自动关机

python train.py;shutdown

使用清华源pip下载安装库

pip install 库名 -i https://pypi.tuna.tsinghua.edu.cn/simple

AutoDL关闭终端或网络程序继续运行的方法

cd到文件运行的路径

输入下列命令:

nohup python -u main.py > /root/autodl-tmp/output.log 2>&1 &

nohup可以将代码放到后台运行,如果需要将打印和输出的内容输出,可以加上-u,>后面的路径是输出log日志的路径,2>&1是将 标准错误 和 标准输出 输出到log日志中。

在命令行或另外的终端的命令行输入以下的命令可以查看实时的输出

tail -f /root/autodl-tmp/output.log

查看进程

ps -ef|grep main.py

杀死进程

kill 进程号