安装torchvision-0.12.0+cu113版本

本来想实现多层感知机的算法y但是需要导入一个d2lzh_pytorch

解决方法:在网上下载了一个d2lzh_pytorch包的文件夹,放在Lib/site-pakeage目录下

                  但是运行的时候还是不行,说缺一个torchtext的包。

结果使用pip安装,可以运行,但是又warning,说torch版本和torchvision、torchtext版本不兼容

pip install torchtext  -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

所以想把torch卸载了,换一个更低版本的,结果不小心把torchvision卸载了。

所以需要把torchvision安装上。

在网站上下载了一个和python版本、cuda版本匹配的whl的文件,网站链接https://download.pytorch.org/whl/torch_stable.html

nvcc -V     或nvcc --version                             命令行查看cuda版本

我的python版本3.8,cuda版本10.2.89所以我选择的是

torchvision-0.12.0+cu113-cp38-cp38-win_amd64

pip install C:\Users\22060\Desktop\torchvision-0.12.0+cu113-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

结果这个命令直接把我的torch1.12版本直接卸载了,并且安装了

torch1.11版本和torchvision的0.12.0+cu113版本。

在这里又出现问题

安装torchvision-0.12.0+cu113版本_第1张图片

oh,崩溃...

然后我选择了更新torchtext的版本,问题解决!并且torch更新为了1.12版本。

pip install --upgrade torchtext -i https://pypi.tuna.tsinghua.edu.cn/simple

出现warning,显示我的加载图像有如下图所示的

torchvision报错:UserWarning: Failed to load image Python extension: Could not find module

解决方法在链接里。(16条消息) torchvision报错:UserWarning: Failed to load image Python extension: Could not find module_fresh_young_man的博客-CSDN博客

安装torchvision-0.12.0+cu113版本_第2张图片

 

然后又发现一个大问题,我的电脑是有独显的,但是跑程序的时候,我发现我的GPU使用率为0,CPU都要干到冒烟了。发现我的torch是cpu版本的。所以这回必须给他卸载了装GPU版本。

先把自己的torch卸载,然后在pytorch的官网上找到对应的版本。GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

然后竟然自带的torchvision版本也更新到了cuda版本,之前更新torch版本的时候,系统自己就将我的torchvision版本由cuda版本更新到了cpu版本,再一次运行程序,oh,终于成功了!

安装torchvision-0.12.0+cu113版本_第3张图片

 

你可能感兴趣的:(深度学习,pytorch,人工智能)