从零开始下载torch+cu(无痛版)

从零开始下载torch+cu(无痛版)

一,前言

由于搞了些新电脑,经常需要配深度学习环境,老是搜来搜去的深感繁琐,干脆记录一下搜自己的老快了hh。

二,配置torch的GPU版具体步骤

一般来说正常的conda,和pip直接下载都会直接寄,或者等会再寄,遂需要直接下载whl然后安装、不排除有通过pip+default-timeout成功的天命之子哈

1,查看电脑安装的Cuda版本

nvcc -V

可以看到我是CUDA11.6

如果没有安装Cuda的可以参考这个博客: 如果,不是N卡,只能。。。使用钞能力了,爱莫能助。

2,在pytoch官网检索待下载whl包名以及版本

pytorch 过往历史版本地址: 举个例子:我查询得到我的cuda版本是11.6那么我们找找cu11.6支持哪些torch版本

然后找到了图上:这个发现有一个有一行#CUDA 11.6下面一行pip。 pip 里面的内容就是我们要手动下载的包。 我的是:

torch1.13.0+cu116 torchvision0.14.0+cu116 torchaudio==0.13.0

大家可视自身情况,安装适合自己cuda的torch,torchvision,torchaudio版本

如果想要的torch版本和自身的cuda的不匹配怎么办?那就卸载cuda重新安装就好了(慎重),这个是重装cuda的教程:

3,下载指定torch,torchvision,torchaudio三个库

地址: 注:记得关掉魔法上网 这里我以下载torch的whl包为栗子 点击去之后, ctrl+f 输入:我的是11.6所以输出cu116 其余版本同理 在第二节中可以知道要下载的torch版本为:torch==1.13.0+cu116 然后我是python10,windows系统,所以我下载这个:torch-1.13.0+cu116-cp310-cp310-win_amd64.whl 注:cp310不是310个对象昂,是指python版本 其他三个同理。

下载如上

可以看到我下载的和pytorch官网里面的,不能说毫不相干只能说是一摸一样。

4,安装进目标深度学习环境中

打开Anconda Prompt 输入:

conda create -n pytorch13 python=3.10
conda activate pytorch13
pip install C:UsersAdminDownloads	orch-2.0.0+cu117-cp39-cp39-win_amd64.whl
pip install C:UsersAdminDownloads	orchvision-0.14.0+cu116-cp310-cp310-win_amd64.whl
pip install C:UsersAdminDownloads	orchaudio-0.13.0+cu117-cp39-cp39-win_amd64.whl

注:pip install 后面的地址变成你们下载的whl绝对地址就好,安装顺序依次是 torch torchvision torchaudio

5,测试是否安装成功

python
import torch
 print(torch.cuda.is_available())

当打印True的时候就恭喜,你配置torch+GPU深度学习环境成功了!

经验分享 程序员 微信小程序 职场和发展