题外话:有些同学的笔记本电脑可能没有显卡或者显卡不太好,建议安装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块钱一小时。

物联沃分享整理
物联沃-IOTWORD物联网 » 安装CPU版本的pytorch

发表评论