pycharm配置pytorch

本文档详细介绍了如何在Windows环境下配置PyCharm以使用PyTorch,包括安装Anaconda、CUDA和CuDNN,创建conda虚拟环境,安装PyTorch,以及在PyCharm中设置Python解释器。步骤清晰,适合初学者参考。
摘要由CSDN通过智能技术生成

pycharm配置pytorch

1.先安装Anaconda

Anaconda3-5.2.0-Windows-x86_64.exe




2.下载cuda

检查电脑是否有合适的GPU
在桌面上右击如果能找到NVIDA控制面板,则说明该电脑有GPU。控制面板如下,并通过查看系统信息获取支持的Cuda版本
在这里插入图片描述

点击 帮助->点击 系统信息 弹出下面的对话框,在驱动程序版本那一栏就能看到该计算机使用的驱动版本。
在这里插入图片描述

下载Cuda
官网:https://developer.nvidia.com/cuda-10.1-download-archive-update2

然后,根据实际情况选择合适的版本
在这里插入图片描述

3.下载CuDNN

官网 https://developer.nvidia.com/rdp/cudnn-download

安装Cuda
1.与安装其他的软件类似
2.安装结束后将 ~/nvcc/bin(因为版本的不同可能在不同的地方) 目录添加到环境变量
3.在命令行下输入 nvcc -V, 出现下列信息说明Cuda安装成功
4.将CuDNN压缩包解压后,下面的三个文件夹复制到Cuda的安装目录下
在这里插入图片描述
在这里插入图片描述

4.配置虚拟环境和安装pytorch

1.首先我们先来为我们的pytorch创建一个虚拟环境。
打开Anaconda Prompt
在这里插入图片描述

在指定目录创建conda虚拟环境:

conda create --prefix=目录 python=3.9

默认安装环境

conda create -n pytorch python=3.8 //这里我使用的是3.8版本的python

选择y就可以等待创建了,安装后输入以下指令来查看安装情况。

conda info --envs

在这里插入图片描述
这个就表示pytorch环境安装完成
2.安装pytorch
接下来就到了安装pytorch的关键时刻了
pytorch官网,选择适合自己cuda的pytorch。
在这里插入图片描述

向下拉动找到这里,选定win-conda-python-你的CUDA版本号
复制下面链接,注意注意注意,可以把-c pytorch去掉,下载更快
//进入pytorch环境
conda activate pytorch
//输入复制的链接,便开始了漫长的等待下载安装
最后,如何检查自己是否安装成功呢
//首先进入pytorch环境,若已进入忽略
conda activate pytorch
python
>>>import torch
//输入完成后如果没有报错,则证明你已经成功安装了恭喜你!!!
>>>torch.cuda.is_available()
true
//继续输入上面代码,输出True,则意味着你可以使用 GPU

5.为你的pychram配置上pytorch

在这里插入图片描述

展开python解释器,选择现有解释器,点击右边三个小点选择第二个conda环境
	一般目录C:\Users\用户名\anaconda3\envs\pytorch\python.exe

这样,整个流程就安装完毕了。

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

pikaqiu one

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

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

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

打赏作者

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

抵扣说明:

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

余额充值