anaconda安装pytorch

添加清华镜像

-添加镜像
清华镜像https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

-显示检索路径

conda config --set show_channel_urls yes

-检查一下看是否添加上了镜像

进入anaconda-environments-channels,应该会有刚才add的镜像,并删除default
在这里插入图片描述

创建自己的pytorch环境

-打开anaconda,进入environments

-点击create,出现如下图所示
在这里插入图片描述
Name由自己定义(此处我命名为Pytorch),Python版本根据自己下载版本设定(会自动选择)

-点击create,等待创建环境

完成后,会有base和Pytorch两个环境

Pytorch安装

-根据自己的安装版本,在pytorch网站寻找代码

进入NVIDIA控制面板,查看CUDA驱动版本(我的是11.1.96)
在这里插入图片描述
进入官网:https://pytorch.org/
在这里插入图片描述
复制代码:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
将复制的代码粘贴到命令行格式下(我修改了版本号11.1),弹出提示,输入y,即可完成安装,显示“done”

测试

-激活pytorch环境: conda activate Pytorch(如果已经在这个环境中,既可以省略此步骤)

-进入python: python

-测试pytorch: import torch
                          import tochvision
在这里插入图片描述
显示为上图,没有报错,就表示安装成功!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值