安装pytorch教程(自用)

1.首先是配置清华镜像源

记事本打开 .condarc 文件(C:\Users\xxx),重写其中的内容。注意https改为http

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

2.接着在Anaconda prompt 命令窗口运行 conda clean -i 清除索引缓存,保证用的是镜像站提供的索引。

3.然后就可以安装 PyTorch啦
进入pytorch官网:https://pytorch.org/

网页下拉,即可看到下图,官网会自动根据你的电脑,显示的即是你可安装的CUDA版本,并给出安装命令。

复制官网提供的命令,并删除-c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
打开 Anaconda prompt 命令窗口,进入你刚刚所创建的环境(我的命名是PyTorch)
conda activate PyTorch

进入环境后输入刚才复制好的命令进行下载conda install pytorch torchvision torchaudio cudatoolkit=10.2 

注意:在测试过程中遇到cuda与pytorch版本不匹配的问题,我是选择下载一个旧一些的pytorch进行匹配,也可以对cuda进行更新。

4.测试
打开 Anaconda prompt 命令窗口,激活环境,输入python,进入python开发环境中

输入import torch
torch.cuda.is_available()
输出True就代表成功了
 

引用:PyTorch 最新安装教程(2021-07-27)_风口IT猪的成长录-CSDN博客_pytorch

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值