CUDA_VISIBLE_DEVICES=1 python your_file.py
或者：在代码里加上os.environ["CUDA_VISIBLE_DEVICES"] = "0"
如果被服务器占用，怎么更换服务器运行的物理块？
本文介绍如何在Python中通过`CUDA_VISIBLE_DEVICES`环境变量来指定GPU设备，以便于运行深度学习或其他计算密集型任务。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
