我的电脑联想拯救者。不是用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博客
照着这个教程最后那一段测试一下就行了