安装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环境创建成功

安装CPU版本的pytorch
激活环境
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块钱一小时。

Original: https://blog.csdn.net/qq_41787357/article/details/122952399
Author: Senescy
Title: 安装CPU版本的pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/706296/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球