Python-随笔

目录

• 去除警告信息
• 指定gpu
• 生成requirements.txt

去除警告信息

import warnings
warnings.filterwarnings("ignore")

指定gpu

CUDA_VISIBLE_DEVICES=1 python train.py

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

生成requirements.txt

当前虚拟环境所有包
pip freeze > ./requirements.txt

当前目录项目所有包
pipreqs ./ --encoding=utf-8 --force