安装CPU版本的pytorch
题外话:有些同学的笔记本电脑可能没有显卡或者显卡不太好,建议安装CPU版本的pytorch。这类同学如果实在需要使用GPU加速训练过程,建议使用在线GPU。
打开Anaconda Prompt
查看环境信息
conda info -e
或者使用下面这一条命令
conda info --envs
创建环境
新建名为torch的环境,python版本为3.6
conda create -n torch python=3.6
再使用查看环境信息命令,可以看到torch环境创建成功
激活环境
activate torch
安装CPU版本的pytorch
使用清华镜像源网站,加快安装速度
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
安装pytorch
conda install pytorch torchvision torchaudio cpuonly
在加载过程中弹出提示y/n,输入 y,确定安装
注意:conda install pytorch torchvision torchaudio cpuonly -c pytorch后面的-c pythoch不要,不然又换回国外源了。
验证是否安装成功
进入python:
python
测试Pytorch:
import torch
如果没有报错,说明Pytorch 已经安装成功
退出python:
exit()
有人问哪里有在线GPU:
1.找导师要 2.免费的有:kaggle竞赛平台;colab;paddle3.付费的就更多了,有autodl平台,3090只用2块钱一小时。