- 博客(4)
- 资源 (1)
- 收藏
- 关注
原创 Pytorch的backward()相关理解
最近一直在用pytorch做GAN相关的实验,pytorch 框架灵活易用,很适合学术界开展研究工作。 这两天遇到了一些模型参数寻优的问题,才发现自己对pytorch的自动求导和寻优功能没有深刻理解,导致无法灵活的进行实验。于是查阅资料,同时自己做了一点小实验,做了一些总结,虽然好像都是一些显而易见的结论,但是如果不能清晰的理解,对于实验复杂的网络模型程序会造成困扰。以下仅针对pytorch 0...
2019-06-19 23:23:04 69967 17
原创 Ubuntu 16.04.1 + NVIDIA driver + cuda 8.0 安装
几个月前安装了CUDA环境,借鉴了网上很多博客。由于帮别人重复安装了多次,所以不如写一个博客整理一下。环境: (不排除,目前的步骤与GPU型号有关)系统:Ubuntu 16.04.1 内核版本为 4.4.0-93-generic (当时查询cuda的官方说明,cuda8.0支持内核为4.4.0)GPU: NVIDIA GTX 960 网上下载的cuda8.0 toolkit 安装包
2017-12-22 15:06:10 596
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人