安装pytorch+配置pycharm解释器(超级详细适合小白)

创建pytorch虚拟环境(GPU版本)


1.进入anaconda的终端窗口


2.查看虚拟环境

conda env list


3.创建虚拟环境

注:可以先换源,否则后面下载很慢,换源可以参考其他博客,这里不在赘述

conda create -n pytorch (python=3.8,可写可不写)


4.创建完成之后,进入虚拟环境

conda activate pytorch


5.进入pytorch官网,找到自己电脑可以接受的配置,复制安装指令

官网:https://pytorch.org/get-started/locally/

安装指令:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

删除“-c pytorch” ,会使用你自己的源安装,否则从pytorch官网上安装。


6.复制安装指令,粘贴到对话框,回车,安装

在安装pytorch的虚拟环境,复制安装指令进行安装(手机开热下载的会快很多或者换成镜像源)


7.安装完成后在Python中配置解释器

File->Settings->Project:pytorch->Python Interpreter


8.输入验证代码,测试环境是否装好

import torch
flag = torch.cuda.is_available()
print(flag)

ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

出现以下结果就是装好了

安装PyTorch和Paddle深度学习环境以及PyCharm IDE,特别是针对初学者和不需要手动配置CUDA/CuDNN的情况,可以按照以下步骤操作: **1. 安装Anaconda:** 首先,访问Anaconda官网下载适用于你的操作系统(Windows、Mac或Linux)的最新版Anaconda。安装过程中可以选择只安装基础环境,因为我们将使用`conda`环境管理工具。 **2. 创建并激活新环境:** 打开命令行或终端,创建一个新的Python环境专注于深度学习任务: ``` conda create -n pytorch_paddle python=3.7 # 如果你想用Python 3.7,替换为你需要的版本 conda activate pytorch_paddle ``` **3. 安装PyTorch:** 对于PyTorch,你可以通过`conda`轻松安装: ``` conda install pytorch torchvision cudatoolkit=10.2 -c pytorch # 根据你的显卡驱动选择适当版本 ``` 注意这里我们假设你已经有一个兼容的CUDA版本(如10.2)并且不需要手动安装cuDNN。 **4. 安装PaddlePaddle:** 安装PaddlePaddle: ``` pip install paddlepaddle cpu_ONLY # 使用CPU版本,如果你想使用GPU,改为`pip install paddlepaddle-gpu` ``` **5. 安装PyCharm:** 去PyCharm官网下载社区版免费版并安装安装完成后,打开PyCharm,在左侧导航栏选择"File" -> "Settings" -> "Project Interpreter",点击"+"添加新的Python解释器。选择你刚才创建的`pytorch_paddle`环境。 **6. 设置PyCharm Python插件:** 为了在PyCharm中切换到Paddle环境,你可以在PyCharm设置里配置Python插件(如Pylance)。在Settings > Languages & Frameworks > Python 中,选择你的`pytorch_paddle`环境作为默认解释器。 **相关问题--:** 1. 如何在PyCharm中查看当前使用的Python环境? 2. 如何在PyCharm中同时支持PyTorch和Paddle项目? 3. 如果我想升级或降级环境,怎么做?
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

WinterWanderer

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值