![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
数据加载
文章平均质量分 51
yangling1035
本人计算机博士在读,平时除了科研工作,也愿意将大量的时间花费在阅读上,常阅读的书籍有哲学、管理、理财类的,在后期我也愿意将所学的知识分享给大家。
展开
-
Linux指定GPU跑程序
Linux指定GPU跑程序一、单卡运行1. 在train.py主程序指定import osos.environ[‘CUDA_VISIBLE_DEVICES’] = gpu_idsos.environ必须在import torch之前,不然程序运行的时候默认使用gpu0来训练程序2. linux终端命令行执行CUDA_VISIBLE_DEVICES=gpu_ids python3 train.py(1)CUDA_VISIBLE_DEVICES=1,2 // 仅使用device 1和 devic原创 2021-11-02 09:14:10 · 8901 阅读 · 0 评论 -
pytorch中DataLoader()过程中遇到的问题
pytorch中DataLoader()过程中遇到的问题原创 2021-04-12 23:26:49 · 654 阅读 · 0 评论