pytorch
台风天赋
Stay hungry;Stay foolish。Stay curiosity~
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
关于pytorch环境配置的补充
pytorch与cuda版本兼容性问题 这个问题主要在复现MAttNets[2]这篇论文时发现 非常致命 环境 cuda = 9.2 pytorch = 0.4.1 cudnn = 7.1.4 问题1描述 Found GPU0 TITAN V which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time, but your PyTorch was compiled with CU..原创 2020-12-22 11:18:46 · 745 阅读 · 1 评论 -
pytorch学习记录
1. 在使用clone()时,比如x = y[1, :].clone() 此时的x不是y的视图,只是一个克隆复制品。此时修改x,并不会影响y;同样的修改y也不会影响到x。 但是如果没有使用clone(),则会互相影响。 使用clone还有一个好处是会被记录在计算图中,即梯度回传到副本时也会传到源Tensor 2. pytorch和numpy一样,内部提供了大量的线性代数方法函数。避免重复造轮子。 3. pytorch中,索引操作不会开辟新内存地址(即使用clone函数的原因),但是普通运算会。 所以,原创 2020-07-25 23:48:53 · 358 阅读 · 0 评论
分享