在anaconda环境配置pytorch及使用pycharm加载对应环境

我已经安装了Anaconda 23.11.0,按照以下步骤在Anaconda环境中使用PyTorch:

  1. 创建虚拟环境(可选):为了隔离不同项目的依赖,建议在Anaconda中创建一个虚拟环境。可以使用以下命令创建一个名为"myenv"的虚拟环境:
conda create --name myenv
  1. 激活虚拟环境:创建虚拟环境后,需要激活该环境才能使用其中的软件包。使用以下命令激活名为"myenv"的虚拟环境:
conda init

然后关闭并重新打开命令行,随后:

conda activate myenv

在这里插入图片描述

  1. 安装PyTorch:在激活的虚拟环境中,可以使用conda或pip命令安装PyTorch。如果您的计算机上有支持CUDA的GPU,可以使用以下命令安装PyTorch和CUDA:
    注:查看CUDA和GPU可以用以下命令:
nvidia-smi

在这里插入图片描述
注:想安装CUDA可以参考:CUDA安装教程(超详细)
我的CUDA安装路径
在这里插入图片描述

安装,从pytorch官网选择自己需要的版本,复制最下面的指令:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

在这里插入图片描述

如果您没有GPU或不打算使用GPU加速,可以使用以下命令安装CPU版本的PyTorch:

conda install pytorch torchvision torchaudio cpuonly
  1. 验证安装:安装完成后,可以使用以下代码验证PyTorch是否成功安装:
import torch
print(torch.__version__)
torch.cuda.is_available()

如果能够成功导入并打印出PyTorch的版本号,则说明安装成功。

使用pycharm加载conda的环境

  1. 下载安装pycharm。注意要下载社区版的才免费。安装都比较常规这里不赘述了,有一个勾选的地方可以都勾选上.
  2. 打开pycharm,创建新工程,新工程设置如下图在这里插入图片描述

注意,Path to conda要选择你的anaconda安装路径下的Script文件夹里的conda.exe,选择好之后,下面的Environment里才会有不同工作环境的选项,比如我按上面的方法创建了一个名为pytorch的环境,(也就是创建上面myenv的那个步骤),并且在里面安装了pytorch,那Environment这里就选择这个名为pytorch的环境。然后点击右下角create就创建完成了。

  1. 点击左边栏的python console,发现已经自动识别了该环境下的python,可以输入上面的import torch等代码检测是否成功。在这里插入图片描述
  • 8
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值