luanhailiang 2023-09-23 00:48:48 +08:00
parent a4af74a7cb
commit 1dab8ae3e1
1 changed files with 1 additions and 2 deletions

3
app.py
View File

@ -14,7 +14,7 @@ from multiprocessing import Process, Event
app = Flask(__name__)
device_count=0
device_count = torch.cuda.device_count()
workers = {}
@app.route('/start/<id>')
@ -191,7 +191,6 @@ def post_request(url, data):
# break
if __name__ == '__main__':
device_count = torch.cuda.device_count()
# cProfile.run('test_function()', 'test_function.profile')
# p = pstats.Stats('test_function.profile')
# p.sort_stats('cumulative').print_stats(100) # Top 10 by cumulative time