pycharm创建环境下载gpu版pytorch

pycharm创建环境下载gpu版pytorch

前言

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

GPU准备:下载CUDA,CUDNN

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

创建pycharm环境

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

进入对应venv的Terminal环境

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

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

下载对应版本的gpu版pytorch

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

验证

// 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~
pycharm创建环境下载gpu版pytorch_第10张图片

你可能感兴趣的:(工具配置,python,深度学习,pytorch,pycharm)