- 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
- 🍦 参考文章:365天深度学习训练营-第11周:优化器对比实验(训练营内部成员可读)
- 🍖 原作者:K同学啊|接辅导、项目定制
一、前期准备:
1.1mac环境安装:
由于是初学者,并且还是mac用户(各种找不到解决方法),所以从安装环境开始学习打卡。
这里使用的是pycharm,在pycharm里导入torch包的时候,发现并没有这个模块,
尝试了用终端配置一个新环境,但是出现了错误,以下是具体的终端命令以及出现的错误。
1.创建新环境
Conda create -n pytorch_env
(pytorch_env是自己设置的环境名字)
2.进入环境
Conda activate pytorch_env
3.在环境中安装pytorch包
Conda install pytorch torchvision -c pytorch
4.出现以下问题:
CondaError: Downloaded bytes did not match Content-Length
- 以下是解决方法(因为mac终端下载速度太慢,转换到pycharm里的终端进行下载):
- 打开pycharm,新建项目:
- 在pytorch官网找到了符合mac版本的命令,并复制(网址:https://pytorch.org/get-started/locally/#mac-from-source)
- 在pycharm的terminal里输入命令,即可配置成功。
- 按照官网检验mac os是否成功配置pycharm的方法
import torch
x = torch.rand(5, 3)
print(x)
输出结果:
1.2设置gpu
import torch
import torch.nn as nn
import matplotlib.pyplot as plt
import torchvision
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)