pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]pycharm安装torch和cuda(在anaconda创建的新环境下)

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全家桶1年46,售后保障稳定

1.问题所在

pycharm中torch和tensorflow好像是有些冲突的,所以我创建了两个conda环境(一个名字叫pytorch,一个名字叫tensorflow),其中pytorch环境中没有tensorflow库,tensorflow环境同理。

现在问题在于每次在Terminal中用pip install torch 后总是cpu版本的

pip install torch 

Jetbrains全家桶1年46,售后保障稳定

import torch
print(torch.__version__)
print(torch.cuda.is_available())

这段代码是看torch到底有没有用到cuda(或者我理解为是否用的是gpu版本),输出为False为cpu版本。

2.安装cuda

这个我觉得可能很多人电脑上已经安装了cuda

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

可以自己在电脑中看一下到底有没有,有了更好,没了接下来讲怎么下载NVIDIA cuda

(1)查看自己应该下载NVIDIA版本

右键“此电脑”——管理——设备管理器——显示适配器
这样就能查到自己的显卡型号了

在这里插入图片描述

 NVIDIA显卡驱动的下载地址:官方高级驱动搜索 | NVIDIA
选择自己合适的型号,开始下载

(2)下载对应的cuda版本

CUDA的下载地址:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
选择合适的型号,点击download即可
在这里插入图片描述

检验安装是否安装正确:win+R→cmd→nvcc -V
如图即为安装正确

在这里插入图片描述 

至此,cuda已经安装在你的电脑里了(如果在安装图中遇到其他问题,可以继续搜搜问题所在,但是这个安装流程一定是没问题的) 。接下来就是安装gpu版本的torch

3.安装torch(在pycharm中的Terminal中,因为我喜欢用这个方式,不喜欢用cmd或者anaconda)

直接打开这个网址https://pytorch.org/get-started/locally/

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

 

如何选择我想大家都应该能看懂,其中我用的Terminal所以Package中选的pip,第一个我看了其他博主都选的Stable,至于最后那个Compute Platform选项因为我的cuda版本就是11.7,所以选了这个(听说版本差不多接近就行 不至于那么严格)。

教大家如何选择最后的Compute Platform(即如何查看自己cuda版本)

桌面右键打开NAIDIA控制面板后,在帮助中找到系统信息,点击组件,蓝色这一条就可以看到CUDA 11.7.57(意思就是11.7版本的)

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

 

 至此学会了查看cuda版本以及下载了对应于cuda的pytorch

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

这时候基本成了,吧最后一栏中Run this Command一栏内容直接复制到Terminal中就行。

4.发现用上述方法安装torch很慢很慢

很好解决,打开网址https://download.pytorch.org/whl/torch_stable.html

进去后找到自己需要的版本,比如我是cuda11.7+python3.7+windows,就选择了下面这个

pycharm安装torch和cuda(在anaconda创建的新环境下)[通俗易懂]

 

cu代表cuda(即选用gpu版本而不是cpu版本,这里一定要看仔细),torch1.11.0版本,这个我觉得没有什么特别的要求,但我喜欢新版本,cp37代表python3.7,win代表windows系统,64代表64位。

下载后我是放在我自己创建的pytorch环境中LIB中site-package中,然后在Terminal中写入下面的代码

pip install D:\anaconda\Anaconda\envs\pytorch\Lib\site-packages\torch-1.11.0+cu113-cp37-cp37m-win_amd64.whl

代码说明:文件放在D:\anaconda\Anaconda\envs\pytorch\Lib\site-packages中,文件名字叫torch-1.11.0+cu113-cp37-cp37m-win_amd64.whl,直接回车就行。

至此问题全部解决,如何判定解决?

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出为

1.11.0+cu113
True

 希望大家能够顺利解决此类问题

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/210012.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

  • 如何成为一名卓越的数据科学家?

    如何成为一名卓越的数据科学家?

    2022年3月4日
    37
  • pycharm激活码最新2022.01(JetBrains全家桶)「建议收藏」

    (pycharm激活码最新2022.01)JetBrains旗下有多款编译器工具(如:IntelliJ、WebStorm、PyCharm等)在各编程领域几乎都占据了垄断地位。建立在开源IntelliJ平台之上,过去15年以来,JetBrains一直在不断发展和完善这个平台。这个平台可以针对您的开发工作流进行微调并且能够提供…

    2022年4月1日
    203
  • origin绘图过程的一些经验

    origin绘图过程的一些经验1.在“帮助->learningcenter”选项卡中可以查找教程及下载APP2.如果不小心将工具栏拖到了屏幕中间或者不小心删除了某个工具栏,可以在“查看->工具栏->重新初始化”里边进行重置工具栏。3.图像数字化(Digitize从图上扣点):工具栏上的位置在“查看(V)”的V字右下边,点击之后选择需要扣点的图片位置,即可打开图片进行扣点或者扣线。需要旋转的可以点击“旋转图像”再点下边出现的微旋按钮将图片旋转,然后移动刚刚打开的图片上的四条线使其对齐坐标轴上下边界,输入坐标轴

    2022年5月31日
    70
  • gbk编码表_河北十一选五五码基本分布图

    gbk编码表_河北十一选五五码基本分布图1.GBK码位分布图2.GBK码位说明GBK亦採用双字节表示,整体编码范围为8140-FEFE,首字节在81-FE之间,尾字节在40-FE之间,剔除xx7F一条线。总计23940

    2022年8月1日
    2
  • Google Earth Engine简介

    Google Earth Engine简介GoogleEarthEngine(GEE)是Google提供的对大量全球尺度地球科学资料(尤其是卫星数据)进行在线可视化计算分析处理的平台,未来地球科学的大杀器啊。链接:https://earthengine.google.com/。注意上Google要费点劲。  关于GEE的入门学习可以参考以下链接(该部分转自无形的风的博客,3ks):GoogleEart

    2022年9月19日
    0
  • acwing-2983. 玩具(计算几何)

    acwing-2983. 玩具(计算几何)计算玩具收纳盒中,每个分区内的玩具数量。约翰的父母有一个烦恼—-约翰每次玩完玩具以后总会将玩具乱扔。他们为约翰准备了一个长方形的玩具收纳盒,用来放他的玩具。但是约翰非常调皮,每次都非常随意的将玩具扔进盒子中,使得所有玩具都随意混在一起,这让约翰难以找到他喜欢的玩具。对此,约翰的父母想出了一个对策,用若干个纸板将收纳盒分隔成若干个分区,这样至少扔到不同分区的玩具之间还是能分开的。下面是一个收纳盒的俯视图示例。1.jpg你的任务是,每当约翰将玩具扔进收纳盒中时,确定每个分区中有多少个玩具。输

    2022年8月9日
    2

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号