PyTorch是一个Python的开源机器学习库。它用于自然语言处理等应用程序。
这个库我安装了3天,期间经历了下载失败、下载成功后发现版本不匹配、下载正确版本后无法正确显示模块等各种问题,试了各种途径下载。昨天终于下载好了,记录一下这个血泪过程。
一、在官方网站下载
官网网址在这里 https://pytorch.org/
根据自己的需求情况选择选项,进行下载。
如果要进行深度学习,最好选择基于GPU的版本(CUDA)
所以要进行pytorch安装,首先要进行的是 CUDA 的安装,可以参看https://zhuanlan.zhihu.com/p/94220564
之后根据CUDA的版本选择合适的Pytorch进行安装,直接复制官网命令到环境运行就可以了
BUT !! 官网安装很容易出现HTTPERROR(网络连接断了)或者下载过慢的问题,这是就需要更换下载源(换成镜像)安装。
二、镜像安装
更换安装原直接参考下面网站,这个网站是换成了清华的源
https://blog.csdn.net/lyj223061/article/details/108639378
然后再键入安装命令(我用的是官网上的命令,没有去掉 -c pytorch, 网上有说-c pytorch 是默认用官网下载的意思,不太清楚哈哈,看官可以都试试哈)
下载过程中遇到的问题们
因为我每次下载都总是出现下载不完整的情况(有的依赖包就是死皮赖脸下到一半就不行了),所以在重复执行安转命令的过程中有很多Warning,大致就是需要你删除特定文件并重启,我按照提示删除对应文件也无济于事。。查找资料后发现可能是以前安装留下的无效缓存太多了,所以直接
conda clean --all
然后一直y就可以了。。(只是我的虚拟环境是刚配了里面没有存什么东西,删除了也不可惜,不知道如果存了不少东西这么干是不是杀敌一千自损八百)
清除完缓存后,再次执行安装命令(一切都飞快了起来)
所以当最后用镜像终于下载完成时,我用 import torch 一测试, 直接报错找不到该模块。
import torch ModuleNotFoundError: No module named ‘torch’
执行了这样一条命令
conda install -c pytorch pytorch
此时再执行
import torch
一切都正常了(即出现>>>正常闪烁,无报错提示)
接下来是检查Pytorch是否在GPU上运行,
torch.cuda.is_available()
返回true 说明一切正常,安装完毕!(如果是false..建议重装,说多了都是泪)