03 2021 档案
摘要:若你已成功安装cuda,却还是用不了。可能是因为: pytorch和cuda版本不对应 nvidia驱动和cuda版本不对应 另外,若pycharm中cuda用不了,而jupyter中可以,那很可能是因为你对应虚拟环境未安装jupyter,导致实际进的是base环境的jupyter(这点巨坑)。
阅读全文
摘要:在开始之前先明确“行轴”和“列轴”的方向如上图。图中“行轴”索引范围为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
阅读全文

浙公网安备 33010602011771号