我在Ubuntu20.04版本安装pytorch的一些坑(cpu版本)

我在Ubuntu20.04版本在pycharm中配置pytorch的一些坑(cpu版本)

最近需要做一些深度学习方面的工作。配置环境一直是我很头疼的问题,去年我在Windows上配置tensflow和Keras花了我两周时间。踩了无数的坑,最后完成也颇为玄学(逃。这次配置pytorch就好多了(花了一个晚上和一个下午大概8小时。Ubuntu yes!)虽然也踩了一些坑。.之前查资料时发现只有一篇关于Ubuntu20.04版本安装pytorch的博客。并且很少关于安装cpu版本的pytorch。所以这里总结一下我配置过程中的一些坑。

1.安装anaconda和Ubuntu

安装Ubuntu这里可以参考(https://blog.csdn.net/xjgao75/article/details/105869699)这篇写的非常详细。
安装anaconda可以参考https://blog.csdn.net/zzq060143/article/details/88042075
建议在虚拟机上安装Ubuntu (就是比较吃内存)
安装后如果键入conda命令表示无此命令。
就更新环境变量

echo 'export PATH="~/anaconda3/bin:$PATH"' >> ~/.bashrc(添加环境变量)
source ~/.bashrc(刷新一下)

https://www.cnblogs.com/veagau/p/11730373.html
然后在anaconda中创建一个pytorch环境。
https://conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html

conda create --name pytorch

然后在环境中安装pytorch

conda install pytorch torchvision cpuonly -c pytorch

注意一定要把上面代码中的-c pytorch去掉,这个的意思是从官网下载,从官网下载非常非常慢 这里可以用清华源

在终端中输入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后输入

conda install pytorch torchvision cpuonly

然后等待
如果你的网不太好,会出现错误。大概意思是你的网络长度不匹配,这意思就是网络断了,而且conda也没有断点续传的功能所以再试一试说不定就OK了
我在安装CPU版本的时候加了清华镜像源的很多网址进去,然后就报错了。后来发现CPU版本和Gpu版本的是一样的网址。如果想删除网址可以参考

https://www.jianshu.com/p/21f35cb966fc

这篇 里面有conda添加源和删除了,浏览源的一些命令。

验证

我安装完了之后出现了


Preparing transaction: done
Verifying transaction: done
Executing transaction: done

如果你安装完发现和网上的博客不一样也别慌(我以为是错误(逃 )
验证方法是在终端输入

python
#进入python环境
import torch 
 print(t.__version__) #d打印PyTorch包的版本信息

然后安装pycharm
打开pycharm
选择环境
详情参见
https://blog.csdn.net/aa3615058/article/details/89339790
装完之后

import torch

一般你输入to他就会自动跳出torch了
如果OK就恭喜你成功配置,开始炼丹之旅。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值