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 Laptop GPU
,安装环境是
CUDA11.1+CUDNN11.1
torch1.9.0+cu111
torchvision0.10.0+cu111
torchaudio==0.9.0
在安装之前先要知道自己的显卡支持什么CDUA版本
右键点击进入NVIDIA控制面板
选择帮助——系统信息——组件
这里可以看出我的笔记本支持的CUDA版本为11.5
或者在命令行搜索cmd,打开命令提示符
输入NVIDIA-smi
就可以看到版本信息
1.下载CUDA
我这里安装的CUDA11.1版本
进入CUDA官网下载CUDA11.1:CUDA Toolkit 11.1 Update 1 Downloads
然后等待下载完成即可
还有其他版本可以下载,自己选择合适的版本下载即可:CUDA Toolkit Archive
配置表:NVIDIA CUDA Toolkit Release Notes
2 .下载CUDNN
进入CUDNN网址:cuDNN Archive
根据自己的系统选择对应的版本
这里需要注册才能下载
登录或者注册完成即可下载
然后进行解压
3.CUDA安装
双击下载的CUDA,点击安装,默认即可
安装完毕
4.安装CUDNN
解压CUDNN
找到安装的CUDA文件夹:NVIDIA GPU Computing Toolkit
将解压的CUDNN放在CUDA中
粘贴成功
打开bin目录复制路径
打开cmd
输入nvcc -V,这样就安装好了cuda了
复制extras的demo文件夹
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
在cmd输入
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
运行bandwidthTest.exe
result=pass说明安装成功了
5.下载pytorch
进入pytorch主页:pytorch
选择更多版本
找到CUDA11.1对应的pytorch版本
# CUDA 11.1
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
这里:
torch1.9.0+cu111
torchvision0.10.0+cu111
torchaudio==0.9.0
可以用pip下载,不过速度很慢,直接在网站上下载速度很快:download.pytorch
在官网上找到对应的版本点击下载即可
注释:cu111表示cuda版本为11.1,1.9.0%2表示1.9.0的版本
cp38表示python版本为3.8的版本,3.9表示Python版本为3.9
win_amd64表示windows的操作系统,有linux和win的操作系统
这个一定要对应不然安装会报错的!!
安装torch
cu111/torch-1.9.0%2Bcu111-cp38-cp38-win_amd64.whl
安装torchaudio
torchaudio-0.9.0-cp38-cp38-win_amd64.whl
安装torchvision
cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-win_amd64.whl
这样就下载好了pytorch环境
6.安装pytorch
如果没有安装环境的话,需要重新安装一个虚拟环境,在anaconda安装一个虚拟环境
conda create -n pytorch(创建的环境名称)
激活环境
conda activate pytorch
进入虚拟环境
本文已经创建直接用conda activate pytorch
安装pytorch环境可以直接在环境中输入官网的下载地址,不过速度很慢,建议直接用操作5进行pytorch下载
pip3 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
进入下载的pytorch路径,复制路径:C:\cuda11.1\pytorch1.9.0
在pytorch环境中输入:cd C:\cuda11.1\pytorch1.9.0
然后用pip install
pip install "torch-1.9.0+cu111-cp38-cp38-win_amd64.whl"
pip install torchaudio-0.9.0-cp38-cp38-win_amd64.whl
pip install "torchvision-0.10.0+cu111-cp38-cp38-win_amd64.whl"
conda list
创建一个.py文件
然后选择自己创建的环境
测试环境是否安装完毕pytorch
测试pytorch是否安装好
import torch
print(torch.__version__)
print('gpu:',torch.cuda.is_available())
import time
import torch##################################################for i in range(1, 10):start = time.time()a = torch.FloatTensor(i * 100, 1000, 1000)a = a.cuda() # a = aa = torch.matmul(a, a)end = time.time() - startprint(end)
运行结果
C:\software\anacondas\envs\pytorch\python.exe C:/Users/qiang/Desktop/python编程开发全栈教程+AIoT人工智能物联网/测试/main.py
1.9.0+cu111
gpu: True
2.2406187057495117
0.328183650970459
0.4589543342590332
0.6902165412902832
0.8932676315307617
0.9731338024139404
说明pytorch安装完成
pytorch环境安装(配置:CUDA11.1+CUDNN11.1+torch.9.0+cu111+torchvision0.10.0+cu111+torchaudio==0.9.0)相关推荐
- PyTorch1.9.1 GPU版本安装(python3.8+pyTorch1.9.1, torch1.9.1/cu111 + torchvision0.10.1/cu111)
目录 一.电脑相关信息 1. 电脑显卡环境: 2. 检查电脑显卡驱动版本: 二.安装Pytorch1.9.1/cu111(GPU版本) 1. 准备:新建虚拟环境 2. 安装pytorch1.9.1/c ...
- PyTorch学习笔记(一):PyTorch环境安装
往期学习资料推荐: 1.Pytorch实战笔记_GoAI的博客-CSDN博客 2.Pytorch入门教程_GoAI的博客-CSDN博客 本系列目录: PyTorch学习笔记(一):PyTorch环境安 ...
- MySQL 系列(一) 生产标准线上环境安装配置案例及棘手问题解决
MySQL 系列(一) 生产标准线上环境安装配置案例及棘手问题解决 一.简介 MySQL是最流行的开放源码SQL数据库管理系统,它是由MySQL AB公司开发.发布并支持的.有以下特点: MySQL是 ...
- 使用Clion开发C语言程序,环境安装配置,保姆级教程
墙裂推荐:C语言学习资源汇总,史上最全面总结,没有之一 C语言开发神器--Clion 首先我们到官网下载安装包 安装包下载后,鼠标双击下载的exe文件,然后点击next 选择安装目录,然后点击next ...
- 使用VS2019编写C语言程序,环境安装配置+代码调试
墙裂推荐:C语言学习资源汇总,史上最全面总结,没有之一 环境安装配置 首先我们要在Visual Studio官方网站去下载安装包 https://visualstudio.microsoft.com/ ...
- 使用VS2019开始第一个C语言程序,环境安装配置+代码实例
C语言文章更新目录 C/C++学习资源(百度云盘链接) 计算机二级资料(过级专用) C语言学习路线(从入门到实战) 编写C语言程序的7个步骤和编程机制 C语言基础-第一个C程序 C语言基础-简单程序分 ...
- Meteor环境安装配置
在本教程中,我们将展示如何在windows操作系统安装Meteor .在我们开始学习使用Meteor 之前,我们将需要NodeJS.如果你还没有安装它,则可以点击下表中的链接. 必须条件 Meteor ...
- 一个电脑的重装到java开发环境安装配置的全过程
刚拿到一台别人用过的电脑.看着c盘爆满,而且用了还是windows7操作系统,强迫症发作马上就准备重装系统. 之前换固态使用wepe制作U盘启动盘装系统的步骤和过程全部忘记的,贼尴尬. 同事都看不过眼 ...
- Mac m1 python软件环境安装配置
Mac m1 python软件环境安装配置 一.安装miniconda 二.安装chrome浏览器 三.安装git 四.安装homebrew 五.安装vscode 选择python解释器 vscode ...
最新文章
- 微软MCITP系列课程
- Eclipse无法打开“Failed to load the JNI shared library”
- 关于CAShapeLayer的一些实用案例和技巧
- winform界面闪退
- 遍历一个ul设置各个li不同的样式
- python excelwriter保存路径_Python和Excel 终于可以互通了!!
- java计算器流程图_帮帮忙:Java小计算器代码,及需求分析.流程图.
- [链接].VGA信号时序
- 前端开发---ppt展示页面评论区展示
- 网络不通时自动重启网卡的脚本
- HDU 4335 What is N?(指数循环节)题解
- 大数据调度工具oozie详细介绍
- 俄罗斯方块C++代码(转载他人代码)
- 360保存html,360极速浏览器怎么保存网页为图片?
- 符号---Alt+数字键
- ASP.NET类计算机专业毕业设计题目选题课题
- 使用cecil 完毕 code injection
- CentOS7设置GRUB系统内核开机选单
- 顺丰快递单号查询API接口调试demo【快递100API接口】
- 网络代理和网络转发的区别
热门文章
- SQL注入 ——sql数据库操作基础(一)
- java mysql报表T加1_MySQL报错: java.sql.SQLException: Column count doesn't match value count at row 1...
- python布尔类型运算_9.python的布尔类型与流程控制
- 简单的网络图片加载工具类
- java给qq发消息_QQ发送消息
- springboot改文件头_SpringBoot图文教程4—SpringBoot 实现文件上传下载
- Python爬虫入门教程 26-100 知乎文章图片爬取器之二
- Android Studio查找错误原因
- 前后端分离与前后端不分离的区别
- hibernate 三种状态的转换