![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
OnTheOurWay
用心,精心,静心
展开
-
nvidia-smi查看到的cuda版本和python虚拟环境中的cuda版本区别答疑(from gpt4)
nvidia-smi查看到的cuda版本和python虚拟环境中的cuda版本区别答疑(from gpt4)原创 2023-12-22 09:51:35 · 855 阅读 · 0 评论 -
pytorch模型转换为android支持的模型
1、模型转换(从gpu到cpu)如过在训练的时候采用的是gpu训练的模型,我们需要保存为cpu格式的模型,如下代码所示:def from_gpu_model_to_cpu_model(read_gpu_pth,write_cpu_pth): device = torch.device('cpu') model=torch.load(read_gpu_pth, map_location = device) torch.save(model,write_cpu_pth) pr原创 2022-03-04 14:38:46 · 3611 阅读 · 1 评论 -
pytorch框架下的数据可视化模块visdom安装使用
1、visdom是什么??visdom是facebook开发的一款可视化工具,使用pytorch训练神经网络时,可以使用visdom来实时对数据可视化。2、安装pip install visdom3、启动服务3.1 如果是本地启动服务:利用下边的命令在cmd中输入即可!python -m visdom.server然后会输出以下所示内容:visdom的默认端口是8097You can navigate to http://localhost:8097 3.2如果是想修改指定端口原创 2022-03-03 10:41:57 · 2967 阅读 · 0 评论 -
pytorch网络结构可视化工具torchvis
0、引言深度学习中,模型结构一般都很复杂,是否有一款好用的工具可以讲模型结构可视化呢?本文介绍一款支持pytorch框架的网络模型可视化工具torchvis。可以将网络结构输出到一个pdf文件中。1、安装过程1.1 首先需要下载graphvis。下载链接,选择自己的系统所对应的版本下载即可。如下图所示,我下载的是最新的。1.2 环境变量设置双击path,进入新建如下图所示。将graphvis的安装路径 加入环境变量。(不知道怎么设置环境变量,可以自行百度)检验是否安装成功命令行输原创 2022-03-02 14:36:45 · 1071 阅读 · 0 评论