win10安装pytorch

视频连接:https://www.bilibili.com/video/BV1hE411t7RN?p=1

1. 创建和激活新的环境

conda create -n pytorch python=3.6
conda activate pytorch

2. 安装pytorch前的准备

2.1 查看设备型号

安装pytorch之前需要安装cuda,调用gpu加速,所以需要知道gpu的型号

1、查看显卡型号
方法一:打开设备管理器->显示配适器,查询到为NVIDIA GeForce GTX 1050
在这里插入图片描述
方法二:打开任务管理器->性能
在这里插入图片描述


2、查看驱动版本
打开NVIDIA控制面板->帮助->系统信息
在这里插入图片描述
查看cuda适配版本
在这里插入图片描述
根据驱动版本选择对应的cuda版本
在这里插入图片描述


2.3 安装cuda

cuda官网链接,翻进去傻瓜式安装
老版本的cuda链接:链接链接2
在这里插入图片描述

输入nvcc -V查看是否安装成功
在这里插入图片描述

删除cuda教程:https://blog.csdn.net/weixin_44606139/article/details/127493438



3. 安装pytorch

4、根据步骤1、2查询到的信息,进入pytorch官网,https://pytorch.org/,选择对应版本,复制指令,进入anaconda对应环境下输入指令安装pytorch

先换源,下载的快

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/

后复制下面指令,注意去掉-c pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2

在这里插入图片描述
报错:
在这里插入图片描述
看别人博客链接,说是因为anaconda不是默认装在C盘才报错
解决方法步骤:找到anaconda文件夹右击打开属性->安全->修改为system
在这里插入图片描述
再一次运行安装指令,之后便安装成功。
在这里插入图片描述

4. 测试

打开pycharm或者就在终端上测试:

import torch
torch.cuda.is_available()  

返回为True
在这里插入图片描述
在这里插入图片描述

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

zdb呀

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值