How to choose GPU if you execute commands in conda env?

wjh720 / QPLEX

88 stars 28 forks source link

Open yyf17 opened 3 years ago

yyf17 commented 3 years ago

Ask a question: If you execute the command in docker to select GPU #1:

NV_GPU=1 docker run ......

How to choose GPU #1 if you execute commands in conda env?

python3 src/main.py --config=......