分享安装pytorch过程以及遇到的一些小问题

    技术2026-08-22  13

    解决通过anaconda安装pytorch时候,出现的cuda版本官网没有的情况

    通过anaconda创建pytorch环境问题1:安装pytorch没有cuda10.0版本问题2:安装进度条不动测试

    在学习深度学习框架pytorch的时候需要进行安装,由于之前使用过tensorflow的gpu版本,所以安装过cuda10.0版本,因此在配置pytorch的时候就不需要重新安装cuda(因为下起来好麻烦),但是在官网上准备安装的时候发现问题,并没有cuda10.0这个cuda版本的选择,弄了半天才解决掉,所以想在这里将遇到的问题解决方法分享出来,希望遇到同样问题的朋友可以试试看能否解决,下面就是安装过程(除cuda10.0)以及安装过程中遇到的问题。

    通过anaconda创建pytorch环境

    首先通过anaconda这个管理工具创建一个专门用于pytorch的环境,打开控制台窗口。输入: conda create --name env_pytorch python=3.7

    其中env_pytorch为环境名称,自己可以更改 python版本也可以根据自己的需求更换。

    然后通过查看已安装环境,看环境有没有安装成功。 conda env list 接下来切换至创建好的环境进行安装: activate env_pytorch

    如图可见,括号里的名称为你当前的环境,并且列表中带有 * 的环境为当前环境。

    问题1:安装pytorch没有cuda10.0版本

    进入官网: https://pytorch.org/get-started/locally/. 根据自己的需求选择相应的选择,我在安装的时候最大的问题就是下面并没有cuda10.0的版本,那么我们直接复制10.1下面的安装命令,并且将它改为10.0

    conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

    然后回车,之后会询问你 y/n ? ,选择 y 即可同意安装。 但是问题2就出现了…

    问题2:安装进度条不动

    会发现在安装pytorch的时候进度条几乎没有在动,一度让人怀疑挂掉,别急,将下载源头更换为清华源即可(会快很多哦),如果不知道怎么更换为清华源,别急,参考以下方法即可(应该不会有问题):

    先查看自己有没有已经安装镜像源,在控制台输入命令: conda config --show

    看有没有显示已经安装的镜像源,如下所示: 如果没有也没问题,执行下面的命令安装pytorch的清华源:

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

    然后再执行以下命令进行安装:

    conda install pytorch torchvision cudatoolkit=10.0 注意:这个命令和之前的还是有区别的,细心的同学有没有发现呀,后面的 -c pytorch 被删掉了,这样就可以了。 接下来就可以体验到你家正常网速的下载速度了。 。。。

    嗷对,还有一点,这样的配置下安装pytorch为了匹配 cuda10.0 而安装的版本为pytorch 1.2.0。

    测试

    等安装完成之后,用控制台打开安装了pytorch的python环境(一定要在安装好的环境下打开哦):

    python

    进入python环境后,执行以下代码测试查看安装版本以及是否可以用gpu加速:

    import torch print(torch.__version__) # 查看pytorch版本 print('gpu:',torch.cuda.is_available()) # gpu加速是否可用

    输出如下结果,那么就大功告成啦! end…

    Processed: 0.012, SQL: 9