reusable_code

最新推荐文章于 2023-03-19 15:38:51 发布

熊乐乐666

最新推荐文章于 2023-03-19 15:38:51 发布

阅读量192

点赞数

分类专栏：编程

本文链接：https://blog.youkuaiyun.com/qq_36601957/article/details/103904616

版权

编程专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何在Jupyter Notebook中指定使用GPU进行代码执行，这对于深度学习等高计算需求的任务至关重要。此外，还提供了图像缩放的实用代码，演示了如何从二维列表转换为一维列表的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

### jupyter notebook 中执行代码指定GPU
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0"

### resize image
def resize_image(ori_dir, des_dir,classes):
for cc in classes:
subclass_dir = os.path.join(ori_dir,cc)
subclass_des_dir = os.path.join(des_dir,cc)
imglist = os.listdir(subclass_dir)
num = len(imglist)
for k in range(0,num):
currentpath = subclass_dir +"/"+imglist[k]
im = Image.open(currentpath)
x_s=256
y_s=256
temp = im
out = temp.resize((x_s,y_s),Image.ANTIALIAS)
out.save(subclass_des_dir+"/"+str(imglist[k]))

ori_path = ' '
des_path = ''
classes = [' ', ' ']
des_path = resize_image(ori_path, des_path, classes)

### 将二维list变为1维list
from itertools import chain
a = [[1,2,3],[4,5,6],[7,8,9]]
b = list(chain.from_iterable(a))
print(b)
# b = [1,2,3,4,5,6,7,8,9]