CUDA_VISIBLE_DEVICES=1 python my_script.py
PyTorch命令行中执行python程序使用指定的GPU
最新推荐文章于 2024-07-16 22:49:14 发布
CUDA_VISIBLE_DEVICES=1 python my_script.py