Pytorch

虚拟环境安装GPU版本的Pytorch

摘要：看了网上很多版本叫你安装gpu版本的pytorch后，自己写了一个比较简单的方法！步骤很少，而且不会出什么问题！不仅在虚拟环境，在你的最初的那个环境也可以这样安装！而且下载速度很快！亲测有效！ torchvision安装也是同理！注意我的是安装了anaconda！安装方法可以参考： Anaco 阅读全文

posted @ 2022-03-25 23:48 yyyyyu 阅读(149) 评论(0) 推荐(0) 编辑

GPU相关：Pytorch

摘要：这是cuda官网,大家自己参考下载方法。 CUDA Toolkit 11.6 Update 1 Downloads | NVIDIA Developer 如果已经下载好，那么可以进行以下流程。 #可以使用nvidia-smi命令来查看显卡信息。 !nvidia-smi 或者在cmd命令行窗口输入以下阅读全文

posted @ 2022-02-27 19:35 yyyyyu 阅读(38) 评论(0) 推荐(0) 编辑

Pytorch读写文件

摘要：所有文件都是保存在pytorch文件的同一个目录下。 1.加载和保存张量 import torch from torch import nn from torch.nn import functional as F x = torch.arange(4) torch.save(x, 'x-file' 阅读全文

posted @ 2022-02-27 19:03 yyyyyu 阅读(33) 评论(0) 推荐(0) 编辑

pytorch小知识（动手学深度学习）

摘要：1.查阅文档 1.1为了知道模块中可以调用哪些函数和类，我们调用dir函数。 import torch print(dir(torch.ones)) 结果如下：通常，我们可以忽略以“__”（双下划线）开始和结束的函数（它们是Python中的特殊对象），或以单个“_”（单下划线）开始的函数（它们通常阅读全文

posted @ 2022-02-24 10:20 yyyyyu 阅读(23) 评论(0) 推荐(0) 编辑

虚拟环境安装pytorch，完整安装过程

摘要：1.提前安装好anaconda 在anaconda prompt命令窗口进入以下命令行 2.创建虚拟环境 conda create --name d2l python=3.8 -y d2l为虚拟环境的名称 3.激活d2l环境 conda activate d2l 4.安装库（python3.8版本为阅读全文

posted @ 2022-02-23 15:51 yyyyyu 阅读(243) 评论(0) 推荐(0) 编辑