【小白安装pytorch环境 cuda cuDNN torch torchvision】

一、安装CUDA

命令行查看驱动版本及可支持的最高CUDA版本

Win+R输入cmd打开命令行,输入命令NVIDIA-smi查看
在这里插入图片描述

查看显卡是否在支持CUDA安装的列表中

参考:https://blog.csdn.net/qq_56794107/article/details/131505821

下载安装

下载cuda地址:https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述

https://developer.nvidia.com/cuda-11.3.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=Server+2016&target_type=exe_local

参考:https://blog.csdn.net/weixin_42984235/article/details/128216440

验证安装是否成功

打开新的控制台,输入nvcc -V
在这里插入图片描述

二、cuDNN下载https://developer.nvidia.com/rdp/cudnn-archive#a-collapse897-118 cuDNN下载网址

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse897-118 cuDNN下载地址
复制链接 迅雷下载
在这里插入图片描述
下载完成后解压,将文件夹下的三个子文件复制到cuda安装路径下的CUDA1下
在这里插入图片描述

三、pytorch安装

  1. 官网下载
    https://pytorch.org/
    在这里插入图片描述
    2.离线下载
    官网下载特别慢,建议离线现在
  • 参考博客:离线下载1
  • 离线包下载网站:https://download.pytorch.org/whl/torch
  • 文件命名规则,如 torch-1.9.0+cu102-cp38-cp38-win_amd64.whl,表示1.9.0版本的pytorch,支持cuda10.2和python3.8,window64位系统
    下载的pytorch版本要和电脑中python的版本相对应,否则无法安装

在这里插入图片描述
3.安装

//	切换到虚拟环境,使用命令安装
pip install [路径]
//	验证安装是否成功,切换到python命令验证
import torch
print(torch.__version__)

在这里插入图片描述
5.安装torchvision

  • 查看版本torchvision
    torch与torchvision对应关系,来源:官方Repo: vision
    在这里插入图片描述
  • 命令行安装
//	切换到虚拟环境,使用命令安装
pip install [路径]
//	验证安装是否成功,切换到python命令验证
import torch
print(torch.__version__)

在这里插入图片描述

附录

torch版本(whl文件)下载地址:

https://download.pytorch.org/whl/torch_stable.html

配置镜像源

如果conda下载过慢,可配置镜像源,
参考博客:https://blog.csdn.net/Supremelv/article/details/132826458?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522171159297616800222850940%2522%252C%2522scm%2522%253A%252220140713.130102334…%2522%257D&request_id=171159297616800222850940&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allsobaiduend~default-4-132826458-null-null.142v100pc_search_result_base6&utm_term=conda%E6%9C%80%E5%85%A8%E9%95%9C%E5%83%8F%E6%BA%90&spm=1018.2226.3001.4187

  • 17
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值