当前位置: 首页 > news >正文

环境配置04:Pytorch下载安装

说明:

显存大于4G的建议使用GPU版本的pytorch,低于4G建议使用CPU版本pytorch,直接使用命令安装对应版本即可

GPU版本的pytorch的使用需要显卡支持,需要先安装CUDA,即需要完成以下安装

1.查看已安装CUDA版本

GPU对应的CUDA版本可以通过以下方法查看:

1.查看当前显卡驱动支持的CUDA版本 打开cmd,输入nvidia-smi

2.查看以安装的CUDA版本 可以通过命令“nvcc -V”或“nvcc --version”

3.如果nvidia-smi命令无效,则说明可能已经安装了cuda 有效解决:nvidia-smi报错Failed to initialize NVML: Unknown Error 且 print(torch.cuda.is_available())一直卡住_nvvm unknown error-CSDN博客

2.下载安装对应版本的cuda

配置显卡cuda与配置pytorch - 知乎 (zhihu.com)

3. python、torch、torchversion版本对应关系及安装

3.1 不建议使用pytorch官方的下载安装方式:

3.2 建议手动下载、手动安装:

1.在download.pytorch.org/whl/torch_stable.html 或download.pytorch.org/whl/torch/ 或阿里镜像源pytorch-wheels安装包下载_开源镜像站-阿里云 (aliyun.com) 中,

2.根据cuda版本、torch版本、python版本,下载对应的 torch.whl文件

 

3.根据torch与torchversion的对应关系,再下载对应的torchversion.whl文件 pytorch,torchvision与python版本对应关系及安装命令_pytorch python版本-CSDN博客

4.下载到本地文件夹,然后通过命令行的形式安装,这样安装可以避免官方安装中的错误

pip install ''xxx''

5.测试GPU版本的pytorch是否成功安装使用

import torchprint(torch.cuda.is_available())

返回true则成功;

6. 如果返回false,则失败,需要优先检查cuda版本与torch版本是否对应,使用命令:pip list,查看torch是否为版本号+cuda版本号:

3.3 建议命令行安装(未亲自测试)

pytorch,torchvision与python版本对应关系及安装命令_pytorch python版本-CSDN博客

3.4 其他博主 Anaconda/conda中安装torch(可部分参考)

conda环境中cuda的安装 - white514 - 博客园 (cnblogs.com)

Anaconda虚拟环境中安装torch + cuda + cuDNN_anaconda安装cuda-CSDN博客

深度学习环境配置:Anaconda、CUDA、PyTorch安装和那些坑儿 | Jin blog (jinhanlei.github.io)

http://www.lryc.cn/news/378481.html

相关文章:

  • 【杂记-浅谈私有地址】
  • Java基础学习-数组
  • 爬虫 pandas Linux Flume Pig填空题
  • Spring框架中哪些地方使用了反射
  • 难辨真假的Midjourney案例(附提示词):适合练手
  • 数据库讲解---(数据库保护)【上】
  • 【Android】【Compose】Compose的简单介绍
  • 对接钉钉Stream模式考勤打卡相关事件的指南
  • CRMEB PRO企业微信通讯录配置
  • 直播新篇章 | 金仓数据库“零距离”探索与知识挑战双重奏
  • List的介绍
  • Uniapp获取具体地理位置
  • Kafka基础教程
  • 视频智能分析平台智能边缘分析一体机安防监控平台打手机检测算法工作原理介绍
  • 辅助构造函数相关学习以及php实现
  • Linux中nginx.conf如何配置【搬代码】
  • Django REST framework序列化器详解:普通序列化器与模型序列化器的选择与运用
  • 红队内网攻防渗透:内网渗透之内网对抗:隧道技术篇防火墙组策略ICMPDNSSMB协议出网判断C2上线解决方案
  • 【Autoware】Autoware.universe安装过程与问题记录
  • JavaScript的学习之旅之基本数据类型
  • oracle开放某些视图给特定用户,查询报视图不存在问题
  • es 更新索引mapping和setting
  • 广告联盟流量变现app开发
  • 【Python高级编程】pickle`文件处理:序列化与反序列化
  • 如何在 Linux 上安装 Docker Desktop
  • 勒索病毒猖狂,请提前做好安全防护,德迅卫士保护你的安全
  • Pycharm怎么默认终端连接远程服务器
  • CTFHUB-SSRF-端口扫描
  • 日期工具类:获取 当前周 | 上一周 | 下一周 的第一天(周一)和最后一天(周天)
  • 大数据学习-Hadoop