Win10+RTX3080Ti显卡安装Pytorch记录

Win10+RTX3080Ti显卡安装Pytorch记录

显卡驱动安装

**方法一:**下载鲁大师安装所需驱动
**方法二:**下载驱动精灵安装所需驱动
**方法二:**进入NVIDIA官网手动搜索驱动程序,然后默认安装

安装Visual C++支持(要使用CUDA,必须要有一个编译工具)

进入Visual Studio官网,选择社区版Visual Studio 安装。(我安装的是VS2017,安装选择里面的Visual C++组件即可,其他按需安装)
在这里插入图片描述
在这里插入图片描述

安装CUDA和CUDNN

Win+R–>运行页面输入cmd输入nvidia-smi命令查看当前驱动所需的CUDA。去CUDA官网下载,默认安装,(不重新安装驱动,我安装的11.5)
在这里插入图片描述
默认安装后,环境变量中自动有CUDA_PATH, 若没有手动新建。
Win+R–>运行页面输入cmd,输入nvcc -V后回车有CUDA相关信息,即安装成功。
在这里插入图片描述
CUDNN官网下载对应版本的CUDNN
在这里插入图片描述
将下载下来的文件解压复制到安装路径中
在这里插入图片描述

安装Pytorch

默认已经安装好anaconda和配置好国内源。

#配置虚拟环境
conda create -n py37 python=3.7
#创建完成后进入环境
conda activate py37
#安装pytorch
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
#pip超时,加上timeout的时间
pip install --default-timeout=1000 torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

为了测试pytorch是否成功安装,在环境下输入python,输入以下代码,返回True则表示安装成功。

import torch
torch.cuda.is_available()

在这里插入图片描述

pytorch下载链接

从官网下载需要的软件包版本

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值