pytorch
cv小白菜
这个作者很懒,什么都没留下…
展开
-
pip安装opencv失败的解决方法
pip安装opencv原创 2022-09-06 09:55:28 · 1927 阅读 · 0 评论 -
pytorch实现多gpu训练(单机多卡)
采用nn.DataParallel实现多gpu训练1、首先选择GPU训练,定义device,“cuda:0"代表起始的device_id为0,如果直接使用"cuda”,同样默认起始为0。device = torch.device("cuda:0" if torch.cuda.is_available() else)2、然后选择GPU的数量if torch.cuda.device_count() > 1: model = nn.DataParallel(model, device_ids=[原创 2021-11-09 09:23:25 · 2269 阅读 · 0 评论 -
服务器系统(Ubuntu)安装及环境(pytorch+gpu)配置(Tesla T4)
1.Ubuntu系统安装1.1 进入启动盘1.2 按e键 添加 “ nomodeset nodmraid xforcevesa” 启动安装1.3 按照实际要求一步一步安装PS:如果到此界面选择键盘后卡死,是由于超微IPMI的显卡与乌班图内置显卡驱动不匹配导致,需要在安装时加上xforcevesa nomodeset参数,然后按实际需求根据屏幕提示安装2. 安装显卡驱动(Tesla t4)注意(PCI设置一下,否则驱动安装不上去)2.1 配置网络例:(文本编辑器)vim原创 2021-11-05 17:32:39 · 2939 阅读 · 0 评论 -
只安装torchvison
pip install --no-deps torchvision原创 2021-10-30 14:28:19 · 392 阅读 · 0 评论 -
pip快速安装scikit-image包
pip install --user --index-url https://pypi.douban.com/simple/ scikit-image原创 2020-05-28 15:46:37 · 2750 阅读 · 0 评论 -
ubuntu16.04 pip配置pytorch 清华源解决超时问题
ubuntu16.04 pip配置pytorch 清华源解决超时问题pip --default-timeout=100 install torch=1.3.0 torchvision=0.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Markdown编辑器, 可以仔细阅读这篇文章,了解一下Markdown的基本语法知识。新的改变我们对Markdown编辑器进行了一些功原创 2020-05-28 15:37:40 · 696 阅读 · 1 评论