Pytorch安装win10GPU(GTX1050ti)

我的电脑联想拯救者。不是用pip、conda装pytorch,直接用清华源来下,速度快,不会出错

2022.3.25更新

最近给女朋友装的时候发现个问题,在第二步中运行conda install pytorch torchvision cudatoolkit=10.1 -c pytorch时,会装pytorch1.10,但是我们可以进入清华源查看一下,pytorch1.10是不支持cuda10.1的,因此必须给定pytorch版本

conda install pytorch=1.7.1 torchvision cudatoolkit=10.1 -c pytorch

然后python的版本也尽量和文章一样,以免出现其他问题

2021.6.18更新

用阿里云下

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

1.CUDA,cudnn

CUDA,CUDnn安装过程照搬下面网址

win10+1050ti 安装 pytorch-gpu_无九不知名的博客-CSDN博客

我下的是cuda10.1,和其相应的cudnn,官网下的慢,我有百度云会员,在下面的网址里面下载

CUDA10.1安装包 - 空眠 - 博客园

2.安装pytorch

我用的anaconda,创建新环境

conda create -n torch python=3.8

激活

conda activate torch

运行

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

 如果上述过程安装成功跳到步骤4,如果上述过程下载不了,看下面的

——————————————————————————————————————————

这个过程一般是下载不了的,很慢,所以运行只是为了看一下你需要下载什么东西(换源啥的我都试过了,没有用),然后按下ctrl+c退出安装就行了

根据上面的提示在清华的镜像源找相应的文件下载,网址在下面

Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

在这里面就快很多。下载完了安装

上面是我的下载的位置,然后一样装了这三个

conda install E:\IDM下载\pytorch-1.7.1-py3.8_cuda101_cudnn7_0.tar.bz2

装的时候可能会报一个错误,因为咱们一开始直接用conda装的会下载一个pytorch文件,可以看一下错误,就是让你手动删除pytorch的一个文件夹,删掉再次运行就行了 

3.再次运行

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

会下载需要的东西,就不会下载那个700M的pytorch了,但是需要下一个300M的,一开始等一下,后面就快了 。

4.测试

win10+1050ti 安装 pytorch-gpu_无九不知名的博客-CSDN博客

照着这个教程最后那一段测试一下就行了

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值