PyTorch:安装和配置
安装
pip安装
pip3 install torch torchvision
macos还需要安装
brew install libomp
否则出错:ImportError: dlopen(/...torch/_C.cpython-36m-darwin.so, 9): Library not loaded: /usr/local/opt/libomp/lib/libomp.dylib
Referenced from: .../torch/lib/libshm.dylib Reason: image not found
安装固定cuda版本对应的pytorch
pip install torch==1.7.1+cu92 torchvision==0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
其它cuda版本对应的pytorch版本查询[https://download.pytorch.org/whl/torch_stable.html]
如cuda=9.0版本时,支持的最高pytorch版本可能就是1.1.0了:cu90/torch-1.1.0-cp36-cp36m-linux_x86_64.whl。
[INSTALL PYTORCH][INSTALLING PREVIOUS VERSIONS OF PYTORCH]
-柚子皮-
显卡驱动版本-cuda版本-pytorch版本需要一一对应,否则torch.cuda.is_available()=False。
显卡驱动版本查询
$nvidia-smi #Driver Version: 384.81
Mon Jan 11 14:51:48 2021
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.81 Driver Version: 384.81 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Tesla P100-PCIE... On | 00000000:04:00.0 Off | 0 |
| N/A 33C P0 25W / 250W | 0MiB / 16276MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
| 1 Tesla P100-PCIE... On | 00000000:81:00.0 Off | 0 |
| N/A 28C P0 25W / 250W | 0MiB / 16276MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
cuda(toolkit)版本查询
nvcc --version或nvcc -V,如果 nvcc 没有安装,那么用
$cat /usr/local/cuda/version.txt
CUDA Version 9.0.176
查看pytorch对应cuda的版本
import torch
print(torch.version.cuda)
查看 cuDNN 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果没有,那么可能没有安装 cuDNN。
Note: NVIDIA cuDNN是用于深度神经网络的GPU加速库。
驱动版本和cuda(toolkit)版本对应关系
CUDA Toolkit | Linux x86_64 Driver Version | Windows x86_64 Driver Version |
---|---|---|
CUDA 11.2.0 GA | >=460.27.04 | >=460.89 |
CUDA 11.1.1 Update 1 | >=455.32 | >=456.81 |
CUDA 11.1 GA | >=455.23 | >=456.38 |
CUDA 11.0.3 Update 1 | >= 450.51.06 | >= 451.82 |
CUDA 11.0.2 GA | >= 450.51.05 | >= 451.48 |
CUDA 11.0.1 RC | >= 450.36.06 | >= 451.22 |
CUDA 10.2.89 | >= 440.33 | >= 441.22 |
CUDA 10.1 (10.1.105 general release, and updates) | >= 418.39 | >= 418.96 |
CUDA 10.0.130 | >= 410.48 | >= 411.31 |
CUDA 9.2 (9.2.148 Update 1) | >= 396.37 | >= 398.26 |
CUDA 9.2 (9.2.88) | >= 396.26 | >= 397.44 |
CUDA 9.1 (9.1.85) | >= 390.46 | >= 391.29 |
CUDA 9.0 (9.0.76) | >= 384.81 | >= 385.54 |
CUDA 8.0 (8.0.61 GA2) | >= 375.26 | >= 376.51 |
CUDA 8.0 (8.0.44) | >= 367.48 | >= 369.30 |
CUDA 7.5 (7.5.16) | >= 352.31 | >= 353.66 |
CUDA 7.0 (7.0.28) | >= 346.46 | >= 347.62 |
[官方对照表][nvidia显卡,驱动以及cuda版本对应查询]
[pytorch版本,cuda版本,系统cuda版本查询和对应关系]
from: -柚子皮-
ref:
PyTorch:安装和配置相关推荐
- Pycharm和Pytorch安装教程配置环境以及遇到的问题:
Pycharm和Pytorch安装教程配置环境以及遇到的问题: 注意:我们每次新建完项目,都要检查一下python解释器和conda.exe是否选择正确. 一.如何找到Anconda哪个环境中安装了p ...
- 【超详细】pytorch安装及配置
文章目录 前言 一.Anaconda安装 二.PyTorch的下载及配置 1.创建虚拟环境 2.在虚拟环境中下载PyTorch 3.本机CUDA版本低怎么下载CUDA高版本的pytorch 4.Jup ...
- PyTorch安装及配置
PyTorch 环境配置及安装 1. 安装 anaconda 2. 为 PyTorch 创建单独的环境,便于管理. 打开 Anaconda Prompt 为 PyTorch 创建单独的环境 pytho ...
- Win 系统 PyTorch 安装配置教程
文章目录 Python 下载配置 PyTorch 安装配置 VSCode 下载配置 Python 下载配置 下载地址:https://www.python.org/ 右键以管理员身份运行 PyTorc ...
- pytorch环境安装(配置:CUDA11.1+CUDNN11.1+torch.9.0+cu111+torchvision0.10.0+cu111+torchaudio==0.9.0)
文章目录 1.下载CUDA 2 .下载CUDNN 3.CUDA安装 4.安装CUDNN 5.下载pytorch 6.安装pytorch 本文的显卡是 NVIDIA GeForce RTX 3060 L ...
- 【CANN训练营第三季】【进阶班笔记3】PyTorch安装与环境配置
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 文章目录 前言 一.pandas是什么? 二.使用步骤 1.引入库 2.读入数据 总结 前言 提示:这里可以添加本文要记录的大概内容: ...
- Pytorch安装(Anaconda配置虚拟环境)(cpu版)
Pytorch安装(Anaconda配置虚拟环境) 最近在学习Pytorch,一开始配好了环境,今天又突然不能用了,只好重配. 之前跟着一本讲Keras和tf的书在Anaconda下配置了tensor ...
- PyTorch安装与环境配置
PyTorch安装与环境配置 NLP课程第一个实验,需要安装PyTorth,下面是一些安装方法及遇到的问题记录. 一.本地安装环境 Windows10 + VScode 网上大多数都是 anacond ...
- 【NVIDIA Jetson NX】 环境配置及pytorch安装
关于CUDA的环境变量导入 jetson nano是原装了CUDA的,但是需要用户导入环境变量(导入相关的路径)才可以使用,只有环境变量导入成功后,方可在命令行使用 nvcc -V 在命令行输入 su ...
- 安装MMDET全过程 历时五天从anaconda安装到pytorch安装到CUDA配置到MIM安装到MMCV安装最后到MMDET安装
深度学习安装笔记(三)--安装MMDET全过程-- 历时五天从anaconda安装到pytorch安装到CUDA配置到MIM安装到MMCV安装最后到MMDET安装 写在前面 安装流程 一.Anacon ...
最新文章
- 自定义算子高性能开发
- php 前台生成多维数组 后台批量添加
- C# 制作外挂常用的API
- BERT论文阅读(一): Pre-training of Deep Bidirectional Transformers for Language Understanding
- php zip 编码,encoding - 用PHP如何检测一个ZIP包内的文件是在何种编码的系统下创建的...
- 你不知道的 Chrome DevTools 玩法
- web.config连接mysql_web.config中配置数据库连接的方式
- kubeadm安装kubernetes 1.13.2多master高可用集群
- 5 呼吸灯verilog与Systemverilog编码
- 餐饮为什么很难做到薄利多销?
- ci php做记录删除,PHP CI APC 使用记录
- 图像融合(四)-- 对比度金字塔
- [转载]for循环的执行顺序
- 当线程在捕捉异常,即使马上杀掉线程,捕捉到异常之后仍然执行异常处理程序...
- html代码学习离线文档,新手学HTML代码的简易方法
- 基于微信小程序的停车位预约系统设计与实现毕业设计毕设开题报告
- 软件工程专业的大三学生经历和感悟
- 什么是关键字驱动框架?
- 单链表的简单操作与演示
- kvm直通sata_KVM虚拟化win10显卡直通一例
热门文章
- (转)(VS2013 )由于应用程序配置不正确,程序未能启动”--原因及解决方法...
- 使用libevhtp搭建HTTPS SERVER(双向验证身份)
- Mac OS Virtualbox 倒入 ova 镜像文件
- BZOJ3223 文艺平衡树
- windows下virtualenv中安装MySQL-python
- WPF 开发 WebBrowser
- Xcode Developer Tools
- Servlet 2.4 规范之第四篇:Servlet上下文
- FLASH+XML:构建简单易更新网站
- 【文末送书】在科研路上,大家有什么经验教训?