- 博客(8)
- 资源 (2)
- 收藏
- 关注
RSS订阅原创 Linux指定GPU跑程序
Linux指定GPU跑程序一、单卡运行1. 在train.py主程序指定import osos.environ[‘CUDA_VISIBLE_DEVICES’] = gpu_idsos.environ必须在import torch之前，不然程序运行的时候默认使用gpu0来训练程序2. linux终端命令行执行CUDA_VISIBLE_DEVICES=gpu_ids python3 train.py（1）CUDA_VISIBLE_DEVICES=1,2 // 仅使用device 1和 devic
2021-11-02 09:14:10
9504
原创 PyTorch实现断点继续训练
当epochs值比较大，在训练过程由于一些原因模型停止训练，好的解决方法是（1）重新训练模型；（2）接着上次训练的断点继续训练一、模型的保存与加载1. 保存整个Moduletorch.save(net, path)2. 保存模型参数state_dict = net.state_dict()torch.save(state_dict , path)二、模型训练过程中保存checkpoint = { "net": model.state_dict(), 'op
2021-11-02 08:41:18
6267
6
原创 Response to Reviewers
#---------------------------------------------------------------------------------------------------------------针对不同的问题有下列几个礼貌术语可适当用用：We are very sorry for our negligence of ………We are very sorry for our incorrect writing ………It is really true as Reviewe
2021-04-16 09:04:57
1124
原创 NVIDIA驱动安装
1. 查看显卡硬件型号在终端输入：ubuntu-drivers devices2. 换源(1) 先备份cd /etc/apt(2) 然后 sudo cp sources.list sources.list.bak(3) https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/ 进入网址选择Ubuntu18.04 (4) 替换sources.list即可(5) 然后sudo apt-get update3. 安装推荐版本(1) 终端输入：sudo
2021-04-15 15:39:45
278
原创 linux使用常见命令
产看磁盘使用情况查看文件夹大小du -h --max-depth=0 * | sort -hrlsblk查看磁盘空间：df -h将新硬盘挂载到home目录下
2021-04-14 17:29:30
199
原创 pytroch的安装及使用
Pytorch的安装pytorch安装遇到的问题安装步骤1. 在linux下安装anaconda3：conda create -n pytorch17 python=3.71. 虚拟环境创建：conda create -n pytorch17 python=3.7pytorch安装遇到的问题安装步骤1. 在linux下安装anaconda3：conda create -n pytorch17 python=3.7备注：一般情况下使用这行代码时朋友torch17将会被安装在anaconda3的envs
2021-04-11 16:08:24
614
split_train_valid.py
2021-04-15
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人