0
点赞
收藏
分享

微信扫一扫

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)


目录

​​查看GPU​​

​​python版本​​

​​安装Pytroch​​

​​匹配版本​​

​​安装PyTorch​​

问题声明

很多python初学者很多都喜欢使用pip直接安装任何东西,但是这里明确一下啊,使用pip安装'torch'几乎是不可能的,我尝试了无数种方案了,即便是看着successfull了,但是使用的时候不能用,气得你半死。可能是我对pip安装torch还是不熟悉吧,但是说明一点,使用pip安装肯定是各种麻烦,所以,直接使用【anaconda】进行安装就好了。

​​anaconda安装详细步骤​​

查看GPU

nvidia-smi

我的CUDA Version:10.1,比较低,没法按照官网上的使用,但是只要不涉及到GPU还是能用其它功能的嘞。

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_pytorch

python版本

python -V

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_官网_02

安装Pytroch

在官网上下载【​​PyTorch​​】

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_python_03

选择【Get Started】也就是开始,在菜单的第一个位置。

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_python_04

匹配版本

我这里CUDA是10.1的,但是没有,所以只能选择11.3的CUDA。选择完毕后可以看到【Run this Command】的提示语句。

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_python_05

安装PyTorch

我这里分开安装了,先安装pytorch再安装torchaudio

conda create -name pytorch python=3.9

比较大,需要安装个半小时左右。 

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_python_06

我这里安装的时候名字写错了,搞成了【ame】

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_官网_07

切换一下环境

conda activate ame

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_pytorch_08

查看安装环境,这里1.12.1就是咱们刚才安装的。

Python安装Pytorch(解决ModuleNotFoundError: No module named ‘torch‘)_pytorch_09

安装【torchaudio】

conda install torchaudio cudatoolkit=11.3

安装完成后通过【pip list】查询一下 



一大堆,OK,安装完毕。能用了。

我这个版本1.12.1,需要根据自己的需求进行该版本。

举报

相关推荐

解决no module named torch

0 条评论