Pytorch 的安装
2023-09-14 09:08:53 时间
GPU版本的安装
- Windows平台 CPU 版本安装
conda install pytorch torchvision cpuonly -c puython
Windows平台需安装VC,需要的联系博主。
- Linux平台 CPU 版本安装
conda install pytorch torchvision cpuonly -c pytorch
- MacOS平台 CPU 版本安装
conda install pytorch torchvision -c pytorch
GPU版本的安装
- GPU版本需有 NVIDIA 显卡硬件支持,请确保您的显卡支持 cuda,具体型号可查询如下网址:https://developer.nvidia.com/cuda-gpus
- 检查显卡驱动
在命令行或者shell中执行:
nvidia-smi
执行完会打印出您的显卡信息:
在右上角,显示您的显卡目前的cuda驱动版本,比如上图显示为11.4,要注意,这里并不是表明您已经安装了cudatookit,仅仅代表,您当前驱动可以支持的 cuda 最高版本。
- 通过pytorch官网选择安装命令,打开pytorch官网(https://pytorch.org/),在网页中间位置,有安装提示栏,如下所示:
这里本人电脑采用如下命令安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
- 检验Pytorch安装完成:
命令行进入 python 使用:
>>> import torch
>>> torch.randn(4,5)
若生成结果,安装完成。
- 验证GPU安装
>>> torch.cuda.is_available()
返回 True 则安装完成。
相关文章
- 关于pytorch安装之后但是没有办法导入pycharm的原因
- RTX3060安装pytorch
- Mac下pycharm如何安装pytorch「建议收藏」
- 从零开始学Pytorch(零)之安装Pytorch
- SQLyog中文版安装教程「建议收藏」
- batchnorm pytorch_Pytorch中的BatchNorm
- pytorch安装详细步骤[通俗易懂]
- PyTorch安装最全流程
- Aarch64 安装Anaconda 和 pytorch
- Keil 2软件下载和安装教程
- 在腾讯云CVM上安装流媒体服务器软件Ti Top Streamer
- 软件管家CAD制图安装包下载AutoCAD 2023软件安装包资源免费下载以及安装教程
- PS下载及安装教程Photoshop全版本下载--所有PS版本都有!
- 如何安装Pytorch
- Windows系统下有英伟达显卡安装PyTorch
- (一)为什么要使用 TypeScript(二)安装并配置 Ts
- 解决eclipse安装maven的问题:Unable to update index for central|http://repo1.maven.org/maven2详解编程语言
- MySQL 免安装版: 快速轻松安装(mysql免安装版安装)
- 快速安装红帽Linux系统:虚拟机安装指南(虚拟机安装红帽linux)
- 开发:使用Linux安装谷歌输入法(linux谷歌输入法)
- 驱动Linux下以太网驱动程序安装指南(linux的以太网)
- 在 Linux 上安装必应桌面墙纸更换器
- MySQL命令行安装指南(mysql命令行安装)
- 如何正确安装Redis服务(怎样安装redis服务)
- 虚拟机环境下Redis安装全攻略(虚拟机下redis安装)
- 2021年MySQL安装指南(2021mysql安装)
- 一键安装,Redis让你更简单(一键安装包 redis)
- 下载 MySQL Zip 版,快捷安装数据库(mysql zip版下载)