离线安装torch和torchvision

编程入门 行业动态 更新时间:2024-10-07 05:19:07

<a href=https://www.elefans.com/category/jswz/34/1767604.html style=离线安装torch和torchvision"/>

离线安装torch和torchvision

一、查看Windows11的cuda版本

打开NVIDIA control panel应用,点击左下角的系统信息

进入后点击组件,再查看cuda

ps:不知道需不需要查看cudnn,我没查看,网上的方法都没用,但是不影响我安装成功。

二、下载对应版本的torch并离线安装

进入任意一个可用的镜像源下载你的python版本和cuda版本对应的torch
我用的是清华镜像源
/

我的python是3.8 cuda11.6,这里我选择的是下面这个包下载


ps:这里需要注意版本对应,torch,torchvison的版本不要出错。可以去PyTorch中torch、torchvision、torchaudio版本对应关系这篇文章中查看。还可以去官方文档1、官方文档2中查看。

下载完成后,打开cmd,进入环境,并cd到文件夹中输如下面代码即可

conda install --offline pytorch-1.12.0-py3.8_cuda11.6_cudnn8_0.tar.bz2
conda install --offline torchvision-0.13.0-py38_cu116.tar.bz2

出现下面两个情况即为安装成功

(base) PS C:\Users\19192> conda install --offline pytorch-1.12.0-py3.8_cuda11.6_cudnn8_0.tar.bz2
Preparing transaction: done
Verifying transaction: done
Executing transaction: done(base) PS C:\Users\19192> conda install --offline torchvision-0.13.0-py38_cu116.tar.bz2Downloading and Extracting Packages
######################################################################################## | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done

更多推荐

离线安装torch和torchvision

本文发布于:2024-03-09 17:43:17,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1725631.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:离线   torch   torchvision

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!