跟着土堆大佬的视频学习pytorch,但是在tensorboard的安装上花费了一些时间,所以写下博客记录一下啦。
刚开始使用终端安装tensorboard的时候,输入以下命令,显示安装成功之后再运行以下代码依旧显示报错No module named ‘tensorboard‘。
pip install tensorboard
from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter("logs")
#writer.add_image()
#y=x
for i in range(100):
writer.add_scalar("y=2x",2*i,i)
writer.close()
后来发现是因为我的终端前面显示的是(PS),不是(pytorch)环境,所以在弹幕中找到了大佬们提供的解决方案:打开设置(setting),然后在工具中找到终端(tool——terminal),修改Shell path:为 C:\Windows\system32\cmd.exe,确定之后再按上述流程,在终端上输入命令,等待安装成功就好啦,运行代码就不会报错啦。