torch和tensorflow
文章平均质量分 68
m0_46973820
这个作者很懒,什么都没留下…
展开
-
在pytorch中自定义dataset读取数据
dataset;dataloader原创 2022-07-30 09:23:28 · 455 阅读 · 0 评论 -
第二步 学习
help(torch.nn.Conv2d);dir(torch)原创 2022-07-29 11:28:20 · 282 阅读 · 0 评论 -
第一步 安装环境
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch 最高支持CUDA11.6 nvcc --version 我安装了CUDA11.2 import tensorflow as tf tf.autograph.set_verbosity(0) physical_devices = tf.config.experimental.list_physical_devi........原创 2022-07-25 23:31:20 · 282 阅读 · 0 评论