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

本文详细介绍了如何在Python 3.6环境中安装与GPU兼容的PyTorch、Torch和Torchvision。推荐离线安装,通过下载对应版本的whl文件,使用pip在conda环境下安装。安装前需确保已安装正确版本的显卡驱动、CUDA和CuDNN。文章提供了国内镜像源以加速在线安装,但未指定GPU版本可能导致安装CPU版本。
摘要由CSDN通过智能技术生成

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

在线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

或者:(第三方扩展包&#x

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值