pytorch环境搭建

已安装好Anaconda环境为前提

1.打开Anaconda Prompt

输入下面代码,查看conda环境下的内容: 

conda info --envs

输入下面代码,新建pytorch环境,并给环境安装python

conda create -n pytorch python=3.8

 切换到新建的pytorch环境中:

conda activate pytorch

2. 安装pytorch准备

(1)进入pytorch官网,https://pytorch.org/ ,根据自己的电脑选择相应配置

(2)更换清华镜像源,依次输入下列代码

 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
 conda config --set show_channel_urls yes
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

 (3)复制pytorch页面的代码,如截图选中部分,我的是:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

去掉后面的 -c pytorch并运行。

(4)安装完成后验证是否安装成功

输入  pip list ,发现torch的版本

输入python,进入python环境。输入import torch 没有报错。

输入 torch.cuda.is_available() ,有返回true / false .

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值