ubuntu 18.04 训练时无法选择GPU

xin376151654 · 发表于 2022-1-31 10:40:06

星级打分

1
2
3
4
5

平均分:3 参与人数:1 我的评分:未评

本帖最后由 xin376151654 于 2022-1-31 12:39 编辑

ubuntu 18.04 训练时无法选择GPU
nvidia-smi是正常的
训练的时候选择GPU的选项也没有了，这个问题有大神遇到过吗

训练时：

bash 6_train_SAEHD.sh

CommandNotFoundError: Your shell has not been properly configured to use 'conda activate'.
To initialize your shell, run

$ conda init <SHELL_NAME>

Currently supported shells are:
  - bash
  - fish
  - tcsh
  - xonsh
  - zsh
  - powershell

See 'conda init --help' for more information and options.
IMPORTANT: You may need to close and restart your shell after running 'conda init'.

Running trainer.

[new] No saved models found. Enter a name of a new model :
new

Model first run.
[0] Autobackup every N hour ( 0..24 ?:help ) :
0
[n] Write preview history ( y/n ?:help ) :
n
[0] Target iteration :
0
[n] Flip SRC faces randomly ( y/n ?:help ) :
n
[n] Flip DST faces randomly ( y/n ?:help ) :
n
[2] Batch_size ( ?:help ) :
2
[128] Resolution ( 64-640 ?:help ) :
128
[wf] Face type ( h/mf/f/wf/head ?:help ) :
wf
[liae-ud] AE architecture ( ?:help ) :
liae-ud
[256] AutoEncoder dimensions ( 32-1024 ?:help ) :
256
[64] Encoder dimensions ( 16-256 ?:help ) :
64
[64] Decoder dimensions ( 16-256 ?:help ) :
64
[22] Decoder mask dimensions ( 16-256 ?:help ) :
22
[y] Masked training ( y/n ?:help ) :
y
[n] Eyes and mouth priority ( y/n ?:help ) :
n
[n] Uniform yaw distribution of samples ( y/n ?:help ) :
n
[n] Blur out mask ( y/n ?:help ) :
n
[y] Place models and optimizer on GPU ( y/n ?:help ) :
y
[y] Use AdaBelief optimizer? ( y/n ?:help ) :
y
[n] Use learning rate dropout ( n/y/cpu ?:help ) :
n
[y] Enable random warp of samples ( y/n ?:help ) :
y
[0.0] Random hue/saturation/light intensity ( 0.0 .. 0.3 ?:help ) :
0.0
[0.0] GAN power ( 0.0 .. 5.0 ?:help ) :
0.0
[0.0] Face style power ( 0.0..100.0 ?:help ) :
0.0
[0.0] Background style power ( 0.0..100.0 ?:help ) :
0.0
[lct] Color transfer for src faceset ( none/rct/lct/mkl/idt/sot ?:help ) :
lct
[y] Enable gradient clipping ( y/n ?:help ) :
y
[n] Enable pretraining mode ( y/n ?:help ) :
n
Initializing models: 100%|################################################| 5/5 [00:01<00:00,  2.50it/s]
Loading samples: 100%|#############################################| 1985/1985 [00:02<00:00, 850.86it/s]
Loading samples: 100%|#############################################| 1985/1985 [00:02<00:00, 831.83it/s]
=========== Model Summary ============
==                               ==
==          Model name: new_SAEHD ==
==                               ==
==    Current iteration: 0       ==
==                               ==
==--------- Model Options ----------==
==                               ==
==          resolution: 128    ==
==          face_type: wf       ==
==    models_opt_on_gpu: True    ==
==                archi: liae-ud ==
==             ae_dims: 256    ==
==             e_dims: 64       ==
==             d_dims: 64       ==
==          d_mask_dims: 22       ==
==    masked_training: True    ==
==    eyes_mouth_prio: False    ==
==          uniform_yaw: False    ==
==       blur_out_mask: False    ==
==          adabelief: True    ==
==          lr_dropout: n       ==
==          random_warp: True    ==
==    random_hsv_power: 0.0    ==
==    true_face_power: 0.0    ==
==    face_style_power: 0.0    ==
==       bg_style_power: 0.0    ==
==             ct_mode: lct    ==
==             clipgrad: True    ==
==             pretrain: False    ==
==    autobackup_hour: 0       ==
== write_preview_history: False    ==
==          target_iter: 0       ==
==    random_src_flip: False    ==
==    random_dst_flip: False    ==
==          batch_size: 2       ==
==          gan_power: 0.0    ==
==       gan_patch_size: 64       ==
==             gan_dims: 16       ==
==                               ==
==----------- Running On -----------==
==                               ==
==       Using device: CPU    ==
==                               ==
======================================