pytorch指定使用某个gpu运行

在python文件第一行加上以下语句，这个python脚本就只会使用id为“2”的gpu。import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "2"# use multiple gpus# os.environ["CUDA_VISIBLE_DEVICES"] = "0

adrianna_xy

2669人浏览 · 2020-05-20 19:48:36

adrianna_xy · 2020-05-20 19:48:36 发布

在python文件第一行加上以下语句，这个python脚本就只会使用id为“2”的gpu。

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
# use multiple gpus
# os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

九章云极普惠算力

更多推荐

ollama使用gpu运行大模型

九章云极普惠算力

成功安装faiss-gpu

九章云极普惠算力

tensorflow-gpu缺少的.dll文件下载：一键解决TensorFlow GPU运行难题

tensorflow-gpu缺少的.dll文件下载：一键解决TensorFlow GPU运行难题【下载地址】tensorflow-gpu缺少的.dll文件下载此仓库为TensorFlow GPU版本用户提供了运行所需的缺失动态链接库文件，如cudart64_110.dll、cublas64_11.dll等。这些文件是...