window10系统安装Python torch模块问题

小鱼儿 2022-11-30 12:57 587阅读 0赞

### 文章目录 ###

*   *  一、说明：
     *  二、解决方法：
     *   *  方法1：官网参考安装命令：
         *   *  采坑：最后发现我的window电脑不支持cuda的GPU，所以需要使用cpu版本的：
         *  方法2：下载文件安装：
     *  三、其他一些报错解决：
     *   *  报错1：\`untimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is\_available() is False. If you are running on a CPU-only machine, please use torch.load with map\_location=torch.device('cpu') to map your storages to the CPU.\`

## 一、说明： ##

今天再GitHub上搜到一个项目，想跑一下，结果需要安装torch模块。然后直接`pip install torch`报错，安装不成功。  
[项目地址：][Link 1]

## 二、解决方法： ##

[pytorch官网][pytorch]  
[Python的torch模块][Python_torch]

### 方法1：官网参考安装命令： ###

安装命令：

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

具体来源：  
安装下面的绿色圈着的，点选之后出现安装命令，cmd下可以安装，但是有点慢。

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjA4MTM4OQ_size_16_color_FFFFFF_t_70_pic_center]

#### 采坑：最后发现我的window电脑不支持cuda的GPU，所以需要使用cpu版本的： ####

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjA4MTM4OQ_size_16_color_FFFFFF_t_70_pic_center 1]

安装命令：

pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

### 方法2：下载文件安装： ###

我这边下载好的whl文件，可以直接安装：  
[下载文件地址：][Link 2]  
去这里下载对应的，然后执行

pip install 下载的绝对理解地址（可以直接拖过来）

## 三、其他一些报错解决： ##

### 报错1：`untimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.` ###

[解决参考：][Link 3]

看到了这一句：

my_model = net.load_state_dict(torch.load('classifier.pt', map_location=torch.device('cpu')))

在原来的基础之上加上这个参数`map_location=torch.device('cpu')`即可  
![在这里插入图片描述][20200825115654127.png_pic_center]

[Link 1]: https://github.com/15950552800/Text_select_captcha
[pytorch]: https://pytorch.org/
[Python_torch]: https://pypi.org/project/torch/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjA4MTM4OQ_size_16_color_FFFFFF_t_70_pic_center]: /images/20221123/ba7570fdf2364428b9efeb10a8522bc2.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjA4MTM4OQ_size_16_color_FFFFFF_t_70_pic_center 1]: /images/20221123/a9480412fcbc4a38aba0b36ef36ea667.png
[Link 2]: https://www.lfd.uci.edu/~gohlke/pythonlibs/#pytorch
[Link 3]: https://stackoverflow.com/questions/56369030/runtimeerror-attempting-to-deserialize-object-on-a-cuda-device
[20200825115654127.png_pic_center]: /images/20221123/afcfb42424734827b18a4ed74255d7bb.png