03 2021 档案

摘要:若你已成功安装cuda,却还是用不了。可能是因为: pytorch和cuda版本不对应 nvidia驱动和cuda版本不对应 另外,若pycharm中cuda用不了,而jupyter中可以,那很可能是因为你对应虚拟环境未安装jupyter,导致实际进的是base环境的jupyter(这点巨坑)。 阅读全文
posted @ 2021-03-29 18:07 rain-fallz 阅读(691) 评论(0) 推荐(0)
摘要:在开始之前先明确“行轴”和“列轴”的方向如上图。图中“行轴”索引范围为0-2,轴长为3,“列轴”索引范围0-3,轴长为4。 对张量的操作 import torch t = torch.tensor([[3,4,1,0],[12,1,2,3],[52,34,88,11]]) t # 输出为: tens 阅读全文
posted @ 2021-03-28 19:29 rain-fallz 阅读(456) 评论(0) 推荐(0)