pycharm配置pytorch-程序员宅基地

技术标签: python  pycharm  windows  

pycharm配置pytorch

1.先安装Anaconda

Anaconda3-5.2.0-Windows-x86_64.exe




2.下载cuda

检查电脑是否有合适的GPU
在桌面上右击如果能找到NVIDA控制面板,则说明该电脑有GPU。控制面板如下,并通过查看系统信息获取支持的Cuda版本
在这里插入图片描述

点击 帮助->点击 系统信息 弹出下面的对话框,在驱动程序版本那一栏就能看到该计算机使用的驱动版本。
在这里插入图片描述

下载Cuda
官网:https://developer.nvidia.com/cuda-10.1-download-archive-update2

然后,根据实际情况选择合适的版本
在这里插入图片描述

3.下载CuDNN

官网 https://developer.nvidia.com/rdp/cudnn-download

安装Cuda
1.与安装其他的软件类似
2.安装结束后将 ~/nvcc/bin(因为版本的不同可能在不同的地方) 目录添加到环境变量
3.在命令行下输入 nvcc -V, 出现下列信息说明Cuda安装成功
4.将CuDNN压缩包解压后,下面的三个文件夹复制到Cuda的安装目录下
在这里插入图片描述
在这里插入图片描述

4.配置虚拟环境和安装pytorch

1.首先我们先来为我们的pytorch创建一个虚拟环境。
打开Anaconda Prompt
在这里插入图片描述

在指定目录创建conda虚拟环境:

conda create --prefix=目录 python=3.9

默认安装环境

conda create -n pytorch python=3.8 //这里我使用的是3.8版本的python

选择y就可以等待创建了,安装后输入以下指令来查看安装情况。

conda info --envs

在这里插入图片描述
这个就表示pytorch环境安装完成
2.安装pytorch
接下来就到了安装pytorch的关键时刻了
pytorch官网,选择适合自己cuda的pytorch。
在这里插入图片描述

向下拉动找到这里,选定win-conda-python-你的CUDA版本号
复制下面链接,注意注意注意,可以把-c pytorch去掉,下载更快
//进入pytorch环境
conda activate pytorch
//输入复制的链接,便开始了漫长的等待下载安装
最后,如何检查自己是否安装成功呢
//首先进入pytorch环境,若已进入忽略
conda activate pytorch
python
>>>import torch
//输入完成后如果没有报错,则证明你已经成功安装了恭喜你!!!
>>>torch.cuda.is_available()
true
//继续输入上面代码,输出True,则意味着你可以使用 GPU

5.为你的pychram配置上pytorch

在这里插入图片描述

展开python解释器,选择现有解释器,点击右边三个小点选择第二个conda环境
	一般目录C:\Users\用户名\anaconda3\envs\pytorch\python.exe

这样,整个流程就安装完毕了。

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_41511783/article/details/124454132

智能推荐

用installshield打包时在注册表里注册项和键值的动态设定方法_注册表new - > key-程序员宅基地

文章浏览阅读1.1w次。在11.5版本里,注册表的函数变化不多我在这里要记录的是打包的时候注册表的问题,注册表分项和键,打开注册表(开始->运行,输入regedit,确定),可以在左边的树形里看见很多文件夹的图标,这些是项,点击一个项,可以在右边看见他的键值,每个项都有一个“默认”,有的项还有很多键值在打包的时候,我们有时候要记录很多东西,嗯,我们要记录安装文件的类型(服务器端程序还是客户端程序),我们要记录软件_注册表new - > key

全志平台lichee启动时间优化_initial_debug-程序员宅基地

文章浏览阅读490次。1. 前言全志平台默认的SDK系统启动+相机出图的时间较长,普遍是十几秒,使用体验较差; 这里尝试通过一些方法优化整体启动时间2.统计内核耗时模块打开这个宏initial_debug,每个驱动的初始化起始时间和结束时间都打印出来了。有了这个时间,基本就可以确定哪些部分需要优化了。 我的做法是只关注耗时10000us以上的驱动。根据上面统计,可根据使用场景的需求, 尝试可以关闭相应不需要的模块3. 降低打印等级开发过程中往往会把系统的打印等级设置为最高,而实际上发布出的固件_initial_debug

《HelloGitHub》第 67 期-程序员宅基地

文章浏览阅读1.5k次。兴趣是最好的老师,HelloGitHub 让你对编程感兴趣!简介分享 GitHub 上有趣、入门级的开源项目。https://github.com/521xueweihan/HelloGi..._开源 lexer

如何做代码评审(code review)_code review怎么做-程序员宅基地

文章浏览阅读5.7k次。Code Review 即日常所说的代码评审或代码回顾,主要是在软件开发的过程中,对功能源代码进行评审,其目的是找出并修正软件开发过程中出现的错误的过程,提高和改进代码质量的过程。_code review怎么做

SSM9==SSM项目启动过程、xml配置SSM项目及需要的3大配置文件、原生SSM未前后端分离的电商网站项目(角色管理员、购买者)只使用了最基础的注解,Model传参_ssm框架 启动传参-程序员宅基地

文章浏览阅读2.1k次。SSM项目的启动过程:ssm框架中,项目启动过程以及web.xml配置详解_菜鸟不会飞-程序员宅基地_ssm项目启动入口是什么本篇主要在基于SSM的框架,深入讲解web.xml的配置web.xml 每个javaEE项目中都会有,web.xml文件是用来初始化配置信息:比如Welcome页面、servlet、servlet-mapping、filter、listener、启动加载级别等。 web.xml配置文件内容如下:<!DOCTYPE web-app PUBLIC..._ssm框架 启动传参

npm设置代理,解决安装不了或安装失败等问题_npm config set disturl 失败-程序员宅基地

文章浏览阅读1.6k次。登陆后不用重启!npm config set proxy http://工号:密码@代理服务器地址:80npm config set https-proxy http://工号:密码@代理服务器地址:80_npm config set disturl 失败

随便推点

MongoDB 的简介、特点以及详解_mongodb特点-程序员宅基地

文章浏览阅读1.8k次,点赞2次,收藏2次。1、概念_mongodb特点

Main class [org.apache.oozie.action.hadoop.HiveMain], exit code [10]_attempt recovered after rm restartmain class [org.-程序员宅基地

文章浏览阅读3k次。oozie执行hiveql,发生这样的异常!Main class [org.apache.oozie.action.hadoop.HiveMain], exit code [10]_attempt recovered after rm restartmain class [org.apache.oozie.action.hadoop

计算机毕业设计springboot网上排课系统的设计与实现w0d059【附源码+数据库+部署+LW】_排课系统数据库设计-程序员宅基地

文章浏览阅读255次。选题背景:随着互联网的快速发展,各行各业都在积极探索数字化转型的道路。教育领域也不例外,传统的排课方式已经无法满足现代学生和教师的需求。为了提高教学效率、优化资源利用以及提供更好的学习体验,许多学校和机构开始引入网上排课系统。这种系统通过在线平台实现课程的安排和管理,为学生和教师提供了更加便捷和灵活的排课方式。选题意义:网上排课系统的出现对于教育领域具有重要的意义。首先,它能够提高教学效率。传统的排课方式需要大量的人力和时间,而网上排课系统可以自动化地完成排课过程,减少了繁琐的手工操作,提高了排课的_排课系统数据库设计

对IO流关闭的思考_c++ 流不关闭-程序员宅基地

文章浏览阅读7.9k次,点赞3次,收藏13次。流必须要关闭的原因java相对C,C++来说不需要手动释放内存,在对象引用被消除之后,正常情况下内存资源是会被垃圾回收,那么在使用完IO流之后为什么需要手动关闭. 这是为了回收系统资源,比如释放占用的端口,文件句柄,网络操作数据库应用等.对Unix系统来说,所有的资源都可以抽象成文件,所以可以通过lsof来观察。看下面这个例子,我们创建许多的IO流但是不关闭public clas..._c++ 流不关闭

无法启动此程序,因为计算机中丢失api-ms-win-downlevel-shlwapi-l1-1-0.dll文件-程序员宅基地

文章浏览阅读2.5w次,点赞2次,收藏4次。最近遇到如题所示弹框的问题,花了很长时间才解决,解决方法却很简单。故记录下来。解决方法:找到相同系统却没有该问题的情况(比如出问题的是64位的win7系统,那就找一个没有该问题的64位的win7系统)。将正常系统的api-ms-win-downlevel-shlwapi-l1-1-0.dll文件复制到出现该问题的电脑的相应位置。就可以解决了。但是同样有一些问题需要注意详细步骤:有一些dl..._api-ms-win-downlevel-shlwapi-l1-1-0.dll

ResponseEntity, @ResponseBody, @ResponseStatus返回Json对象数据用法区分_responseentity json-程序员宅基地

文章浏览阅读325次。ResponseEntity:code:/*常用*/@GetMapping("/hello")public ResponseEntity<String> hello(){ return new ResponseEntity<>("hello word!",HttpStatus.OK);}/*设置HTTP头*/@GetMapping("/hello")pu..._responseentity json