pycharm创建环境下载gpu版pytorch

pycharm创建环境下载gpu版pytorch

前言

本篇文章是根据自己调试的过程所做的总结,主要目的是为了用gpu去跑深度学习的模型,所以需要下载pytorch的gpu版本,但是当中出现了很多错误,所以这里给出一版经试验成功的解决方案。

GPU准备:下载CUDA,CUDNN

这个主要是在官网上找到对应版本,许多之前的博主的方法都是一致的:
第一步,是在桌面点击右键会出现NVIDA控制面板的选项,点击会出现下面的界面。
在这里插入图片描述
第二步,点击系统信息查看自己的显卡对应的CUDA版本如下图所示,我的CUDA版本是11.4。
在这里插入图片描述
第三步,进入NVIDIA官网,下载CUDA以及配套的CnDNN,CUDNN是专门用于配置显卡深度学习的一个库,非必要但是一般来说都会使用这个库,具体的建议自己去进一步了解,或者跟我一样选择一起下下来。
详细下载过程参见以下
链接
也可以寻找其它博客,OK这步完成之后,我们开始创建一个pycharm环境

创建pycharm环境

最开始,我想用anaconda环境去配置,结果半天没搞明白,所以我的解决方案是生成一个项目下的环境,即生成完后会有venv文件夹这种。如下图所示,点击pycharm左上角的setting,点击Python Interpreter.再点击右上角的Add Interpreter,再点击Add local Interpreter.
在这里插入图片描述
OK,接下来会出现下图,选择Virtualenv Environment,选择New,选择对应版本的python点击创建就ok啦。
在这里插入图片描述
在环境创建完之后,首先要解决进入环境的问题.

进入对应venv的Terminal环境

这个是我遇到的问题,那就是当我用pip list指令去测试的时候发现terminal一直在根环境,并没有进入当前的venv文件夹下的环境,很烦~。OK,还是继续给出具体步骤
第一步:Terminal用cd进入venv\Scripts路径一定要进入这个路径哈,如下图所示,这就是我这个项目的路径,
其中cd \可以从当前目录退到根目录,即从C\file跳到C
然后 cd f: 可以帮助你进入另一个盘符,这两个指令够你达到具体的位置了。
在这里插入图片描述
ok,
第二步,执行.\activate指令,进入虚拟环境,我到这第一次就出问题了:(。如果没有那你很棒棒哦。下图为我的错误。
在这里插入图片描述

那么这个问题如何解决呢?进入Windows PromptShell,我的是直接在Pycharm里就有,如果没有去找一怎么进入这个Shell就行了。
在这里插入图片描述
进入Windows PromptShell,进入后输入set-executionpolicy remotesigned,结果我又出错了,不慌,让我们读读低下的错误提示。
在这里插入图片描述
按照提示输入Set-ExecutionPolicy -Scope CurrentUser。出现下图的显示:跟我一样输入RemoteSigned就行了。搞定!
在这里插入图片描述

下载对应版本的gpu版pytorch

进入pytorch官网:
链接
然后就是经典每个博客都有的图片了.对应版本的话,我是CUDA11.4,经验证CUDA11.3是适配的。
在这里插入图片描述
我是点左下角下的之前的版本,但是我傻了,上面已经给了下载的命令啊。。。。不要学我哈。Anyway负责一下命令,回去先.\activate激活环境,然后输入复制的命令,静待佳音,下不了的话科学上网或者清华源吧。如下图所示,激活成功,我用pip list尝试了一下。
在这里插入图片描述
最后一步复制指令,等待接结束就行了。在这里插入图片描述

验证

// A text block
import torch
print(torch.cuda.device_count())
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
print(torch.__version__)

成功后的效果如下图所示了:祝大家成功,愿世界无bug,Peace~
在这里插入图片描述

  • 1
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值