2021-03-22

指定gpu

import os
#多块使用逗号隔开
os.environ['CUDA_VISIBLE_DEVICES'] = '2'

-1表示禁用gpu

cpu转gpu

1. .cuda()变成.cpu()
2. torch.load(args.model_path,‘cpu’)

下载包

要用install下载torch torchvision 不能用conda

sys.path.append()

sys.path 返回的是一个列表！

该路径已经添加到系统的环境变量了，当我们要添加自己的搜索目录时，可以通过列表的append()方法；

对于模块和自己写的脚本不在同一个目录下，在脚本开头加sys.path.append(‘xxx’)：

./和…/以及/之间的区别

1、./是当前目录

2、. ./是父级目录

3、/是根目录

argparse.ArgumentParser()

https://blog.youkuaiyun.com/lizhiyuanbest/article/details/104975848
https://blog.youkuaiyun.com/vcvycy/article/details/88600134?utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.control&dist_request_id=&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.control

Python format 格式化函数

https://www.runoob.com/python/att-string-format.html