设置CUDA_VISIBLE_DEVICES的方法
设置CUDA_VISIBLE_DEVICES的方法
1. 通过bash脚本设置CUDA_VISIBLE_DEVICES
export CUDA_VISIBLE_DEVICES=0,1,2,3
# 注意bash脚本中不要随意使用空格，特别是符号之后不要加空格
source ~/anaconda3/bin/activate torch
# 使用source命令读取并且执行activate，激活anaconda虚拟环境
python -c 'import torch; print(torch.__version__); print(torch.cuda.device_count())'
# 用python command模式显示pytorch版本
2. 通过os.environ来设置CUDA_VISIBLE_DEVICES
import torch
import os
os.environ['CUDA_VISIBLE_DEVICES']='0,1,2'
print(torch.cuda.device_count())