深度学习环境配置

对于入门深度学习的新手来说,配置环境是件让人头疼的事情,今天总结一下Linux服务器配置环境的流程。

1.安装cuda和cudnn

这个之前博客有写过,见下面链接

Linux无root无sudo权限在用户目录安装cuda9.2和cudnn7.1

2.安装anaconda

# install
conda create -n myenv python=3.5 anaconda
# activate env
conda activate myenv 
# 可能需要配置~/.bashrc

细节可以参考下面的博文

安装anaconda

下面在激活的myenv中安装各种包

3.安装opencv

conda install opencv

4.安装PyTorch和torchvision

PyTorch官网选择对应的版本即可
在这里插入图片描述

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

5.安装Tensorflow和TensorboardX

最新的PyTorch已经支持了Tensorflow的Tensorboard,不过我还是在用Pytorch1.0,所以可视化用的是TensorboardX。

conda install tensorflow-gpu
pip install tensorboardX

6.TensorboardX可视化的具体介绍

(1)Python代码

from tensorboardX import SummaryWriter
# 在当前目录生成log文件夹,里面保存可视化的文件
writer = SummaryWriter('log')

# 每条命令是一张曲线图,三个参数分别是曲线名,纵坐标,横坐标
writer.add_scalar('train/loc_loss', loss_l.item(), epoch*epoch_size+iteration)
writer.add_scalar('train/conf_loss', loss_c.item(), epoch*epoch_size+iteration)
writer.add_scalar('train/total_loss', loss.item(), epoch * epoch_size + iteration)

(2)可视化

登陆服务器

连接ssh时,将服务器的6006端口重定向到自己机器上来:

ssh -L 16006:127.0.0.1:6006 username@remote_server_ip

其中:16006:127.0.0.1代表自己机器上的16006号端口,6006是服务器上tensorboard使用的端口。

启动tfboard

在服务器上使用6006端口正常启动tensorboard:

tensorboard --logdir=./log --port=6006

本地浏览器查看

在本地浏览器中输入地址:

127.0.0.1:16006

可视化结果如下图
在这里插入图片描述
参考

7.其他有用的命令

数据迁移

scp (-r) 本地用户名@IP地址:文件名1 远程用户名@IP地址:文件名2

[本地用户名@IP地址:] 可以不输入,可能需要输入远程用户名所对应的密码.

# 把服务器的文件file迁移到本地
scp -r spectre@10.10.10.10:/mnt/home/hhh/file ./ 
创建软链接

一般习惯把数据统一存放在服务区~/data 目录,然后如果用到对应的数据,比如coco,在对应的文件夹里面创建软链接。

# 创建软链接
# ln  -s  [源文件或目录]  [目标文件或目录]
ln -s ~/data/coco data

# 删除软链接 注意不要./coco/,这样会把根目录下的数据删掉
rm -rf ./coco
jupyter notebook

jupyter入门教程(上)

jupyter入门教程(下)

远程访问jupyter notebook

让jupyter运行于指定的conda虚拟环境

程序后台运行

(1)screen让你服务器程序后台运行

(2)nohup后台运行+保存日志

nohup train.py >train.log 2>&1 &
Linux常用命令

linux简单方便的命令集锦之统计与增删

linux简单方便的命令集锦之查看磁盘空间与文件大小及数量

vim常用命令

vim简单方便的命令集锦

Conda和cuda环境移植(克隆)的远程方案

1.cuda和cudnn
直接复制,修改bashrc文件
2.conda
新环境安装conda
复制旧conda文件,然后conda create -n BBB --clone ~/path即可创建新的env BBB。
其中BBB是新建的conda env,~/path是旧环境的路径,如 ~/anaconda/env/det/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值