简单测试

sudo  docker pull taesungp/pytorch-cyclegan-and-pix2pix

sudo nvidia-docker run -it -p 8097:8097  taesungp/pytorch-cyclegan-and-pix2pix bash



cd /workspace/pytorch-CycleGAN-and-pix2pix

bash datasets/download_pix2pix_dataset.sh facades

cd  /workspace/scripts 

vim train_pix2pix.sh

增加:--gpu_ids 0,1,2   --batch_size 2  -niter 100000

python -m visdom.server &bash scripts/train_pix2pix.sh

一般情况下 visdom会报错,或者长时间没有反应



cd /miniconda/lib/python2.7/site-packages/visdom/
 vim server.py
注掉:download_scripts()
#download_scripts()


之后运行 ,看GPU状态
这里运行可能会出现 ,port used的错误。

lsof -i:8097

kill PID(pid下面的那个号)

之后再python -m visdom.server &bash scripts/train_pix2pix.sh
应该就没问题




 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值