| |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
-> Python知识库 -> pycharm安装torch和cuda(在anaconda创建的新环境下) -> 正文阅读 |
|
[Python知识库]pycharm安装torch和cuda(在anaconda创建的新环境下) |
1.问题所在pycharm中torch和tensorflow好像是有些冲突的,所以我创建了两个conda环境(一个名字叫pytorch,一个名字叫tensorflow),其中pytorch环境中没有tensorflow库,tensorflow环境同理。 现在问题在于每次在Terminal中用pip install torch 后总是cpu版本的
这段代码是看torch到底有没有用到cuda(或者我理解为是否用的是gpu版本),输出为False为cpu版本。 2.安装cuda这个我觉得可能很多人电脑上已经安装了cuda 可以自己在电脑中看一下到底有没有,有了更好,没了接下来讲怎么下载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 检验安装是否安装正确:win+R→cmd→nvcc -V ? 至此,cuda已经安装在你的电脑里了(如果在安装图中遇到其他问题,可以继续搜搜问题所在,但是这个安装流程一定是没问题的) 。接下来就是安装gpu版本的torch3.安装torch(在pycharm中的Terminal中,因为我喜欢用这个方式,不喜欢用cmd或者anaconda)直接打开这个网址https://pytorch.org/get-started/locally/ ? 如何选择我想大家都应该能看懂,其中我用的Terminal所以Package中选的pip,第一个我看了其他博主都选的Stable,至于最后那个Compute Platform选项因为我的cuda版本就是11.7,所以选了这个(听说版本差不多接近就行 不至于那么严格)。 教大家如何选择最后的Compute Platform(即如何查看自己cuda版本)桌面右键打开NAIDIA控制面板后,在帮助中找到系统信息,点击组件,蓝色这一条就可以看到CUDA 11.7.57(意思就是11.7版本的) ? ?至此学会了查看cuda版本以及下载了对应于cuda的pytorch 这时候基本成了,吧最后一栏中Run this Command一栏内容直接复制到Terminal中就行。 4.发现用上述方法安装torch很慢很慢很好解决,打开网址https://download.pytorch.org/whl/torch_stable.html 进去后找到自己需要的版本,比如我是cuda11.7+python3.7+windows,就选择了下面这个 ? cu代表cuda(即选用gpu版本而不是cpu版本,这里一定要看仔细),torch1.11.0版本,这个我觉得没有什么特别的要求,但我喜欢新版本,cp37代表python3.7,win代表windows系统,64代表64位。 下载后我是放在我自己创建的pytorch环境中LIB中site-package中,然后在Terminal中写入下面的代码
代码说明:文件放在D:\anaconda\Anaconda\envs\pytorch\Lib\site-packages中,文件名字叫torch-1.11.0+cu113-cp37-cp37m-win_amd64.whl,直接回车就行。 至此问题全部解决,如何判定解决?
输出为
?希望大家能够顺利解决此类问题 |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 | -2024/11/15 14:19:37- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |