Anaconda+pycharm+pytorch安装教程(GPU版)

配置环境

os:win11

Anaconda:3

pycharm:2022.3.2 (Professional Edition)

CUDA:11.7

注意:显卡必须是英伟达且显存6G及以上为佳,否则Pytorch安装在GPU下的运行效果远不如CPU,AMD显卡暂不支持安装。

安装步骤

一、Anaconda:

  1. 官网下载:Anaconda | The World's Most Popular Data Science Platform

  1. 安装:步骤不再具体叙述。

  1. 新建环境

在Anaconda中找到并打开Anaconda Navigator,点击->Environments->点击左下角Create,输入新建环境Name并选择python版本->点击Create,等待一会儿环境即新建完成。

4.环境激活

打开Anaconda prompt->输入activate pytorch->回车(这里的pytorch是刚才新建的环境名,需要根据自己的实际定义修改)。

二、CUDA:

  1. 官网下载:https://developer.nvidia.com/cuda-toolkit-archive

最新推荐下载11.6或11.7版本。

2.安装结果确认

键盘按win+R->输入cmd->回车->输入nvcc -V,正常返回版本即为安装成功。

三、pytorch:

  1. 官网下载:Start Locally | PyTorch

选择相应配置后Ctrl+C复制Run this Command里的内容->打开Anaconda prompt并进入刚刚新建激活好的环境内->Ctrl+V粘贴Run this Command里的内容->回车即可下载并安装;

注意这里的Package选Pip、Compute Platform需注意与安装的CUDA版本保持一致,其他项根据自己的实际情况选择;

到了这里即是已经安装完成。

2.安装结果确认:先输入python->再依次输入下面的三行代码,返回对应结果即为正常。

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

四、pyCharm:

1.环境加载:文件->设置->项目:pythonProject->Python解释器->全部显示,查看是否有新建的Conda环境;

如无则按以下步骤操作:点击左上角+号->添加本地解释器->Conda环境,找到Anaconda安装目录中的_conda.exe文件->加载环境;

注意这里的使用现有环境需选择Anaconda->envs->pytorch;

最后在python解释器项选择刚加载的Conda环境并点击应用,等待环境加载完成点击确认即可;

  1. 安装结果确认:在新建项目中输入次输入下面的三行代码并运行,返回对应结果即为安装成功,至此所有步骤完成。

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

  • 4
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值