Pytorch CPU版本安装教程

        刚换过电脑,今天想重新装一下pytorch的CPU版本,也遇到了诸多问题,这里分享一下本人的安装过程。

        首先默认大家已经安装了anaconda,打开anaconda prompt后,输入如下代码,可获得当前

conda config --show

环境的配置信息。运行

conda info -e

可查看anaconda所有虚拟环境。

       下面是清华镜像源的配置文件.condarc 

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

        创建虚拟环境,当然环境名称和python版本自己决定,遇到y/n都选y。

conda create -n pytorch python=3.10

        然后激活并进入环境。

conda activate pytorch

        安装pytorch,在官网选择cpu版本的pytorch。复制底下生成的代码到anaconda ptompt中。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

        安装完成后 ,输入python,然后输入import torch,返回下一行证明安装完成。

        接下来配置jupterbook的kernel,继续在pytorch这个新环境下进行。

conda install ipykernel
python -m ipykernel install --user --name=enviname  --display-name showname 

其中enviname是新环境名称,showname是jupterbook中kernel的名称,自己定。

新建notenook后,import torch就不会报错了。 

 

    

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年6月8日
下一篇 2023年6月8日

相关推荐