在线or离线安装

最近使用pytorch跑深度学习模型，需要用到与gpu相对应的包，写此博文，提醒后来者避避坑

pytorch

本次使用python3.6（conda环境），一般conda环境可以创建多个Python版本，互不干扰，比单一Python要好用，在pycharm上添加pytorch库。

点击左下方 + ，进去页面输入pytorch，安装对应的版本

torch和torchvision

torch，torchvision库在pycharm中没搜到，在conda环境下激活python36，使用pip命令安装
在使用计算机的狗之前，要安装好显卡驱动，cuda和处cudnn，请至
https://blog.csdn.net/cashmood/article/details/105081586

因为要是用gpu跑代码，所以和torch，torchvision相关的包都必须是和gpu相对应的，

（1）离线安装（强烈推荐）

下载对应的torch和torchvision ：http://download.pytorch.org/whl/torch_stable.html

或者：（第三方扩展包）https://www.lfd.uci.edu/~gohlke/pythonlibs/#pyhook
或者去官网下载对应的包
比如：torch-1.8.0+cu111-cp36-cp36m-win_amd64.whl
torchvision-0.9.0+cu111-cp36-cp36m-win_amd64.whl

我的计算机是rtx3060的显卡，对应的duda版本是11.1（cu111），python36对应cp36, 64位操作系统

下载到本地后：进入conda环境激活对应的python36
conda env list
activate python36

pip install+ whl文件位置
比如：pip install D:\下载（电影，图片）\torch-1.8.0+cu111-cp36-cp36m-win_amd64.whl

（2）在线安装

进入conda环境激活对应的python36，
pip install torchvision
pip install torch
一般torch这个包比较大，从官网下载速度特别慢，还时不时地中断。

还可以指定国内镜像地址
例如： pip install --index-url https://pypi.douban.com/simple
比如: pip install --index-url https://pypi.douban.com/simple torchvison

但是，在线安装不指定具体的包名，一般会安装CPU版本的torch，torchvision，所以不推荐在线安装。

下附国内镜像常用地址
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

各位大佬，看到觉得有用给个赞，嘿嘿！！！

pytorch，torch，torchvision的gpu版本安装避坑相关推荐

Pytorch：入门指南和 PyTorch 的 GPU版本安装(非常详细)
Pytorch: 入门指南和 PyTorch 的 GPU版本安装(非常详细) Copyright: Jingmin Wei, Pattern Recognition and Intelligent S ...
PyTorch1.11.0 GPU版本安装（python3.8+pyTorch1.11.0，torch1.11.0+cu113 torchvision0.12.0+cu113）
目录一.电脑相关信息二.安装Pytorch1.11.0 GPU版本 1. 准备:新建虚拟环境 2. 安装pytorch v1.11.0:(在pytorch官网的历史版本里找安装命令) 电脑:win ...
TensorFlow GPU 版本安装个人总结：Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0
TensorFlow GPU 版本安装个人总结:Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0 接触机器学 ...
YOLO V5 CPU版本安装踩坑记录
YOLO V5 CPU版本安装踩坑记录单纯的记录下yolov5进行检测遇到了点问题的解决方法安装过程: 准备: 1,下载YOLOv5 github项目:https://github.com/ult ...
【鸿蒙开发】hpm-cli的安装避坑、详细使用教程
[鸿蒙开发]hpm-cli的介绍.安装避坑.详细使用教程文章目录 [鸿蒙开发]hpm-cli的介绍.安装避坑.详细使用教程安装环境准备使用教程踩坑集合 HPM包管理器是华为鸿蒙推出的Harmo ...
医学影像阅读/分析软件FSLeyes安装避坑+核磁共振影像数据处理
医学影像阅读/分析软件FSLeyes安装避坑+核磁共振影像数据处理讲解视频内容请移步Bilibili: https://space.bilibili.com/542601735 入群讨论请加v ho ...
pytorch GPU版本安装 win10
我原先装的是pytorch cpu版本,在复现论文源码的时候提示AssertionError: Torch not compiled with CUDA enabled,查了一下资料,又看了一下电脑的 ...
GPU版本安装Pytorch教程最新方法
目录步骤第一步:安装 Anaconda 和 Pycharm 软件第二步:下载安装CUDA11.3 (1)首先查看自己电脑GPU版本方式一:搜索框输入nvidia,打开nvidia控制面板方式 ...
pytorch的cuda环境搭建（GPU版本安装）
第一步:安装cuda 安装cuda,这里还需要安装对应版本的Visual Studio(具体参考本博主的博客 https://blog.csdn.net/qq_36653505/article/det ...

pytorch，torch，torchvision的gpu版本安装避坑

pytorch，torch，torchvision的gpu版本安装避坑

在线or离线安装

pytorch

torch和torchvision

pytorch，torch，torchvision的gpu版本安装避坑相关推荐

最新文章

热门文章