![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习框架安装
文章平均质量分 85
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 · 8836 阅读 · 0 评论 -
pytroch的安装及使用
Pytorch的安装pytorch安装遇到的问题安装步骤1. 在linux下安装anaconda3:conda create -n pytorch17 python=3.71. 虚拟环境创建:conda create -n pytorch17 python=3.7pytorch安装遇到的问题安装步骤1. 在linux下安装anaconda3:conda create -n pytorch17 python=3.7备注:一般情况下使用这行代码时朋友torch17将会被安装在anaconda3的envs原创 2021-04-11 16:08:24 · 527 阅读 · 0 评论