Win10+GPU版Pytorch1.1安装的安装步骤
作者:Taosiyu 发布时间:2023-11-01 16:52:48
安装cuda
更新nvidia驱动
打开GeForce Game Ready Driver或在GeForce Experience中下载符合自己gpu的程序。
选择cuda
打开nvidia控制面板
点击帮助、点击系统信息、在点击组件在3D设置中可以看到cuda信息
在我升级过nvidia驱动后,cuda的版本更新到了10.1。接下来下载cuda .
cuda10.1安装完毕。
安装cuDNN
在安装了cuda10.1后选择对应的cuDNN版本v7.6.1
解压文件,然后添加bin目录到环境变量
安装cuDNN
在安装了cuda10.1后选择对应的cuDNN版本v7.6.1
解压文件,然后添加bin目录到环境变量
使用Anaconda prompt创建环境
conda create -n pytorch_gpu pip python=3.7
其中pytorch_gpu是环境的名字,python版本是3.7
conda activate pytorch_gpu
这个命令可以进入创建好的环境。
安装Pytorch
建议切换源,我使用的是中国科学技术大学”(USTC Open Source Software Mirror)提供anaconda的镜像源,速度提高不少。
方法一:
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
方法二:
打开.condarc文件,进行编辑
这个顺序存在一个优先级,会先在中国科学技术大学的镜像源寻找,如果大家感觉速度慢,也可以换别的镜像源。
打开官网选择后复制命令
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
整个过程只需要按一下'y'即可。
验证
为了确保PyTorch是否安装成功,我们需要运行简单的样例代码测试,例如打印出随机生成的张量矩阵,以及gpu是否可以使用。
import torch
x = torch.rand(5,5)
print(x)
输出类似下面
tensor([[0.7078, 0.1424, 0.3411, 0.3987, 0.3476],
[0.7534, 0.7137, 0.3489, 0.4226, 0.3640],
[0.4104, 0.8411, 0.5112, 0.0629, 0.0664],
[0.7568, 0.9495, 0.3300, 0.2392, 0.6441],
[0.7615, 0.1883, 0.6001, 0.9663, 0.3313]])
如果ok,pytorch安装成功。
torch.cuda.is_available()
返回True,GPU版Pytorch安装完毕。
将PyTorch导入Pycharm
点击Pycharm的选项栏中的File选择Settings,然后点击Project Interpreter,接着点击右侧的Add。
然后在Conda Environment中选择Existing environment
选择Anaconda3下的python.exe,Conda executable选择conda.exe
如果失败,选择Anaconda3下的env下的新创建的虚拟环境中选择python,试一试。
Pycharm开启科学计算模式(Scientific Mode)
具体步骤:
Settings –> Tools –> Python Scientific > Show plots in tool window 勾选
View 勾选 Scientific Mode
Run–>Edit configurations…
勾选 Run with python console
开启科学运算模式,舒服,正在习惯这个姿势。
ps:如果想恢复到普通模式,只需要在view下,取消scientific mode勾选。
结束语
环境安装完毕,以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
来源:https://www.cnblogs.com/taosiyu/p/11443377.html
猜你喜欢
- MySQL支持大量的列类型,它可以被分为3类:数字类型、日期和时间类型以及字符串(字符)类型。本节首先给出可用类型的一个概述,并且总结每个列
- 如下所示:#文件内容 lisilock = open("lock_info.txt", "r+",e
- Python在用GPU跑模型的时候最好开多进程,因为很明显这种任务就是计算密集型的。用进程池好管理,但是tensorflow默认情况会最大占
- 本文实例讲述了Python编码类型转换方法。分享给大家供大家参考,具体如下:1:Python和unicode为了正确处理多语言文本,Pyth
- 阅读上一篇:W3C优质网页小贴士(二) 注意字体大小网页设计者中有这么一种倾向:他们认为小字体能让网页看起来更漂亮,并能提供更多空间给每个网
- PlaywrightPlaywright 是微软公司开源的一款 UI 自动化测试工具,它的第一个版本 v0.10.0 发布于 2020 年
- 1. document.form.item 问题 (1)现有问题:现有代码中存在许多 document.formName.item(&quo
- 本文实例讲述了python创建关联数组(字典)的方法。分享给大家供大家参考。具体分析如下:关联数组在python中叫字典,非常有用,下面是定
- 0.引言利用Dlib官方训练好的模型“shape_predictor_68_face_landmarks.dat”进行68点标定,利用Ope
- 如: 0.625 取 1 2.1 取3 3.6 取4 <% if fix(a)>a then b=fix(a) else b=f
- 1. 滤波器1.1 什么是滤波器滤波器是对图像做平滑处理 的一种常用工具。平滑处理即在尽可能地保留原图像信息的情况下,对像素值进行微调,使邻
- 就目前互联网上大小网站而言,大部分都是采用ASP+ACCESS/SQL Server或者PHP+MySQL来编写;事实上,ASP和MySQL
- 由于工作中涉及到生日编辑资料编辑,然后自己改了一下代码:<html><head> <meta charset=
- 在一些网页应用中,就比如在投票系统中,当我们进行的是多项投票时,我们要求用户最多只能选择几项进行投票,这也是就是说选择复选框的个数最多几个.
- 一、所需工具**Python版本:**3.5.4(64bit)二、相关模块opencv_python模块sklearn模块numpy模块dl
- 例题取用登录模块:代码如下def login_check(username,password):''' 登录校验的函
- 一、文件内容的分发 应用场景:分批读取共有358086行内容的txt文件,每取1000条输出到一个文件当中# coding=utf-8# 分
- 基础知识-----黄金分割法960px宽度的网格设计的好处无需我多说了,下面主要是复习一下如何作图求出960px宽度下的黄金分割点1、首先沿
- 一、背景交通大数据是由交通运行管理直接产生的数据(包括各类道路交通、公共交通、对外交通的刷卡、线圈、卡口、GPS、视频、图片等数据)、交通相
- 如下所示:import sysfrom PyQt5 import QtGui, QtCore, QtWidgetsfrom PyQt5.Qt