在Pycharm中安装Pytorch教程总结

序言

第一次安装pytorch遇到各种问题,然后结合几篇热门的文章总结以下步骤。

一、安装Anaconda

推荐使用清华源下载

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

根据自己电脑的环境选择版本,我这里是windows的就选择这个了,不用选择最新版的Anaconda,具体啥版本可以自主选择。

下载好之后就打开安装包,根据提示一路点下去就行了,默认安装C盘,可以自己选择安装到其他盘。

下载好后开始菜单中可以看到这样,就基本上安装成功了。【我目前没有在系统变量中添加conda路径,暂时感觉也用不着,如果想添加的话也可以搜其他的文章,方法比较简单】

二、打开Anaconda Prompt

conda info -e  // 可以用来查看目前有的虚拟环境

刚开始应该只有一个base的环境,这里的pytorch12.8是我后创的

下一步就需要创建一个新的虚拟环境来安装pytorch,因为好像只能用python3.9的版本,低于3.9的版本就不适用了。

conda create -n xxxxx python=3.9 

通过conda创建一个虚拟环境,3.9是python的版本,都可以按自己需求改,一定要指定具体 python 版本。这里把“xxxxx”随意替换成你想取的名字即可

安装成功后,可以再查看已有的环境

conda info -e

下一步,激活环境

conda activate xxxxx

当前面从(base)变为(pytorch12.8)时表示此时已经切换到你所创建的pytorch虚拟环境,随后正式进入安装pytorch环节。

三、正式安装PyTorch

需要注意的是,使用前应确定自己是需要cpu般的还是gpu版的pytorch,我这里需要用gpu般的所以,后续步骤都以gpu为基础。

  1. 打开pytorch官网:PyTorch,点击Get Started

注意这里CUDA的版本号,每个人都要根据自己电脑的GPU适配的型号去选择

那么如何查看gpu适合什么版本呢?

(1)Win+R  cmd

这里能看到我电脑CUDA版本是12.9的,根据其他的文档,我们在下载CUDA的时候不能高于自己的这个适配版,比如我这里12.9,安装CUDA只能<12.9,所以我安装的是12.8版的。

(2)安装CUDA

PS:看别的安装教程上没有这步,然后当时没装,就没运行起来,后来又搜了半天......

网址:CUDA Toolkit Archive | NVIDIA Developer

刚进来就是上面这样,然后根据自己电脑环境配置依次选择即可,最后点击download下载

下载完成后直接打开文件,无脑安装即可,默认也是C盘,我这里改安装在了E盘。

安装好菜单栏中会多出这些,可通过nvcc -V命令查看cuda版本:

nvcc -V

(3)安装pytorch

pytorch的本质是三个库,torchtorchvisiontorchaudiotorch最大有1G左右,另外两个只是补充的附件,所以工程中我们一般只import torch即可。

直接到其官方网站寻找下载资源,根据需要的版本生成指令

失败经历:这是其他文章中的图,如今打开官网已经没有Conda选项了【下图是最近的官网界面】

但我依照一些文章指导,依旧选择了Conda安装,但是会提示安装失败,原因是找不到包之类的。

从错误信息来看,Conda 无法在当前配置的频道中找到 cudatoolkit=12.8 这个包。这可能是因为你的 Conda 配置中没有包含提供 CUDA 12.8 包的频道,或者 CUDA 12.8 包尚未在这些频道中提供。

为了解决这个问题,你可以尝试以下几个步骤:

1. 更新 Conda

首先,确保你的 Conda 是最新版本,因为较新的版本可能已经解决了这个问题。

conda update conda

2. 添加 NVIDIA 频道

NVIDIA 提供了 CUDA 包,你可以尝试添加 NVIDIA 的 Conda 频道。

conda config --add channels nvidia

3. 搜索可用的 CUDA 版本

在添加了 NVIDIA 频道后,你可以搜索可用的 CUDA 版本,看看是否有 12.8 版本。

conda search cudatoolkit

从这里能看出来最高只有11.8的了

如果 Conda 安装仍然存在问题,可以考虑使用 pip 来安装 PyTorch。可以试着使用以下命令:下面有清华源下载方式,没有vpn的可以用那个!!!

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

【注意】这里下载由于没有找国内源,必须挂梯子,不然网速过慢很多包都会因为超时安装失败

看到这已经安装成功了!!!

【注】清华源下载

pip3 install torch torchvision torchaudio --index-url https://pypi.tuna.tsinghua.edu.cn/simple

(4)简单测试pytorch是否正常

安装完成后输入conda list查看所有库,出现如下图说明安装成功,conda相关命令详细可见常用命令

conda list

在环境中打开python,查看能否导入,示例如下则说明安装成功:

四、Pycharm中配置有pytorch的环境

1.打开设置

2.添加python解释器

刚开始没有添加python解释器,点击右侧的Add Interpreter

3.选择环境

失败经历:其他文章中是添加的Conda环境,我一开始也尝试了

我自己电脑上用conda环境配置完之后没有用,代码中还是提示没有torch包,通过询问得知,这个是版本问题,不同版本有区别,只要能加载到环境就行。

所以我觉得干脆最方便的就是Type直接选Python,只要能加载到环境就行了

步骤:

选择python.exe后选择打开即可

然后点击OK

最后配置好就类似这种,Apply亮着的话就点一下应用也没毛病,最后再点击OK关闭这个界面就好了

然后Pycharm就会开始加载新的环境设置,等待蓝条消失即可

测试一下,torch包有效就结束了。

### 如何在 PyCharm 终端安装 PyTorch #### 创建或选择 Python 解释器 为了确保 PyTorch 安装在一个合适的环境中,在 PyCharm 中应先创建或选择一个 Python 解释器。这一步骤可以通过 PyCharm 的设置功能完成,从而指定项目使用的特定版本的 Python 或者虚拟环境[^1]。 #### 新建并激活虚拟环境 建议为 PyTorch 创建一个新的虚拟环境来隔离依赖项。通过这种方式可以避免不同库之间的冲突,并保持项目的整洁有序。一旦选择了新的解释器或者是已有的虚拟环境,则可以在该环境中安全地安装 PyTorch 及其相关组件而不影响全局 Python 设置[^2]。 #### 使用清华镜像源加速下载过程 考虑到网络条件可能会影响软件包的获取速度,推荐更换国内较为稳定的清华大学开源软件镜像站作为 pip 的索引 URL 来加快下载速度。对于 Windows 用户来说,具体命令如下所示: ```bash pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/ ``` 此命令会永久更改默认的 pypi 源至清华镜像站点[^3]。 #### 执行具体的 PyTorch 安装指令 当一切准备就绪后,就可以直接利用预先下载好的 whl 文件来进行安装了。以下是针对 Windows 平台的具体命令实例;请注意这些命令适用于 CUDA 11.6 和 Python 3.7 版本组合的情况: ```bash pip install torch-1.13.1+cu116-cp37-cp37m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ pip install torchvision-0.14.1+cu116-cp37-cp37m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ pip install torchaudio-0.13.1+cu116-cp37-cp37m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ ``` 上述命令应当依次输入到 PyCharm 内置终端中执行,以确保所有必要的 PyTorch 库都被正确加载到了当前工作空间内。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值