2022年4月5日
指定运行代码的GPU
摘要： 指定用第二个GPU运行代码： import os os.environ['CUDA_VISIBLE_DEVICES']='1' 或者在命令行终端输入的命令前加上CUDA_VISIBLE_DEVICES= * CUDA_VISIBLE_DEVICES=0 python your_file.py 阅读全文
posted @ 2022-04-05 15:02 鱼儿冒个泡 阅读(98) 评论(0) 推荐(0) 编辑