1. 安装pytorch踩坑记录
网上完整步骤的安装贴已经有不少,说下重点解决的,直接使用官网指令安装pytorch11.3发现安装的是CPU版,GPU不能用,含泪找解决步骤。
网上有人说用查看有无cpuonly的方法,试过不管用,具体参见链接安装GPU版本pytorch
解决办法是卸载相关pytorch安装文件,重新安装,使用指令
pip uninstall torch
发现只能卸载由pip安装的torch, 原来由conda安装的依然在,当然也可以搜**使用pip卸载pytorch,很多条指令,逐个库下卸载*,不要觉得没必要,因为直接一条指令卸载卸不干净,已经试过。
还是要老老实实的用conda卸载,时间巨久,我也就是花了十几个小时吧。。。
conda uninstall pytorch
conda uninstall libtorch
这次发现用conda卸载了一堆依赖库,嗯,这次大概可能已经卸载干净了吧?
老老实实的用whl自己下载安装包下载,参考安装GPU版本pytorch,今天网络太差了,pytorch官网登不上,等会再继续写。
换了清华源,参考博客pytorch加速下载——清华镜像源,因为我早就改过,具体步骤就不说了,重点是添加临时下载源,
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
接下来,我的cuda是11.3,所以就安装相应GPU版本的pytorch了,换清华源提示找不到,还是老老实实官网下载吧。
具体下载什么版本,请查阅网址
pytorch以前版本
注意此处需要查阅自己cuda版本,网上教程很多,就不再细说了。
我的是cuda11.3, 查到11.3对应的代码,
要是没有翻墙的话,就替换下载源为清华源或者阿里源, 删掉最后的 -c pytorch
,否则又是官网下载了。这样安装的是pytorch cpu版本,栓Q。不要上来就调用conda指令安装!
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1–extra-index-url https://download.pytorch.org/whl/cu113
这块网址没翻墙的话记得替换为国内源网址,一定记得换。
换源教程参考链接GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~
设置清华源加速安装:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
开始指定版本安装:
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
亲测用pip 指令安装的是pytorch GPU 版本,这一步一坑的实在难绷。要用pip指令安装!!!!
安装完后测试代码,切到python指令模式,输入指令
import torch
torch.cuda.is_available()
显示true成功了!