摘要: import torch #查看torch版本 print(torch.version) #查看cuda版本 print(torch.version.cuda) #GPU是否可用 print(torch.cuda.is_available()) #返回gpu数量 print(torch.cuda.d 阅读全文
posted @ 2022-04-12 11:15 Touming 阅读(6172) 评论(0) 推荐(0)
摘要: python 在使用pip安装的时候,一些小一点的还好,安装一些大的包的时候,会非常的慢,这时我们就可以使用清华大学的镜像来安装,打开cmd pip install XXX -i https://pypi.tuna.tsinghua.edu.cn/simple 比如说要安装selenium包,可以这 阅读全文
posted @ 2021-12-18 20:37 Touming 阅读(426) 评论(0) 推荐(0)
摘要: 国内github镜像:用hub.fastgit.org代替github.com 阅读全文
posted @ 2021-12-17 14:04 Touming 阅读(1222) 评论(0) 推荐(0)
摘要: 搭建LeNet5网络模型(Model.py) 点击查看代码 import torch import torch.nn as nn class LeNet5Model(nn.Module): def __init__(self): super(LeNet5Model, self).__init__() 阅读全文
posted @ 2021-12-12 21:02 Touming 阅读(459) 评论(0) 推荐(0)