基本教程
下载安装GPU版的Pytorch网络上已经有很多教程,Anaconda+PyTorch这里推荐b站的我是土堆up主的安装教程视频,简单易懂,跟着一步步走大体上是没有问题的
但!!!
视频是2022年的,2023年pytorch更新版本后有些许的不一样,下面分享下我这次下载安装遇到的问题和解决办法
问题:python-cuda镜像源找不到文件
按照官网指令下载网络太慢了,所以我选择镜像源下载,这里我选的北京外国语大学的镜像网站
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/
但是终端显示有packages conflicts,类似这样
Specifications:
- torchaudio -> python[version='2.7.*|3.5.*|3.6.*|>=2.7,<2.8.0a0|>=3.5,<3.6.0a0|3.4.*']
Your python: python=3.9
解决办法
pytorch更新后cudatooklist改成了pytorch-cuda,要在nvidia官方网站下载,镜像网站没有及时更新,所以要在后面加上nvidia官网下载源,输入下面的指令就可以找到正确的版本
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/ -c nvidia
或者先在channels里添加好镜像网站,再直接输入下面的指令也可以成功
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c nvidia
这样系统在镜像网站找不到的包就可以去nvidia搜索下载
问题:镜像网站只推送cpu版本的pytorch
这个问题搜索了很多解决办法,最后解决的办法参考了这篇博文,就是需要先在终端输入官网的指令
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
等待一会儿看要下载的pytorch,torchvision和torchaudio的版本,我这里是pytorch-2.0.1, torchvision-0.15.2, torchaudio-2.0.2, cuda-11.8
所以我输入下面的指令就可以了,但镜像网站还是偶尔会发疯,大家多尝试几次说不定就好了
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c nvidia
问题:下载过程中HTTP出错
这个重新再输入一遍一样的指令,或者关了终端重新打开,多尝试几次应该就可以,就是一个连接的问题
这些就大概是我遇到的问题和解决办法,其实配置好镜像源后安装速度挺快的,也很顺畅,主要就是注意一个Anaconda和cuda的版本问题,Anaconda建议选2022年的,毕竟现在已经2023年了,下载安装之前也更新一下自己的conda,版本太老也会出现些不必要的麻烦
还要注意有时候用镜像源安装的包可能会少,在下载之前检查一下所需内存和官网下载所需内存是不是一样的
最后祝大家都能顺利下载安装!