ubuntu16.04系统pytorch安装
2021/11/12 7:13:21
本文主要是介绍ubuntu16.04系统pytorch安装,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
首先创建虚拟环境
conda create -n torch python=3.6(指定默认python为3.6)
激活torch
source activate torch
安装torch
Pytorch官方网站:https://pytorch.org/
安装时需要注意torch和torchvision的对应版本,如下所示
根据自己的系统和python已经cuda版本进行选择安装
官网命令安装
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio cudatoolkit=10.2 -c pytorch
注意这里安装时注意看安装的包是否包含cpuonly,如果包含的话,是无法调用GPU的,其次你如果需要用到自动混合精度训练包(amp)你需要安装PyTorch 1.6以上的版本,不然是不包含amp的
import torch.cuda.amp as amp
清华镜像安装(提高安装速度)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio cudatoolkit=10.2 -c pytorch
本地离线安装
镜像下载地址:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
下载安装的镜像文件
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.8.1-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2 wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/torchvision-0.9.1-py36_cu102.tar.bz2 conda install --offline pytorch-1.8.1-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2 conda install --offline torchvision-0.9.1-py36_cu102.tar.bz2
这里推荐官网命令安装和清华镜像安装(提高安装速度),离线安装可能会出现环境配置问题
安装好后,可以通过一下代码确认torch是否安装成功
import torch a = torch.Tensor([1.]) a.cuda() # 如正常则返回"tensor([ 1.], device='cuda:0')" from torch.backends import cudnn # 如正常则静默 cudnn.is_acceptable(a.cuda()) # 如正常则返回 "True"
通过以下代码看安装的pytorch是GPU版本还是CPU版本,输出false则安装的是CPU版本的,true则是GPU版本
import torch print(torch.cuda.is_available())
安装时如果遇到一下问题:
Fetching package metadata ....... CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://nanomirrors.tuna.tsinghua.edu.cn/anaconda/cloud/linux-64/rpodata.json> Elapsed: - An HTTP error occurred when trying to retrieve this URL. HTTP errors are often intermittent, and a simple retry will get you on your way. ConnectTimeout(MaxRetryError("HTTPSConnectionPool(host='nanomirrors.tuna.tsinghua.edu.cn', port=443): Max retries exceded with url: /anaconda/cloud/linux-64/repodata.json (Caused by ConnectTimeoutError(<requests.packages.urllib3.connecton.VerifiedHTTPSConnection object at 0x7fb6d340dcc0>, 'Connection to nanomirrors.tuna.tsinghua.edu.cn timed out. (connct timeout=9.15)'))",),)
请检查
gedit ~/.condarc
将里面的东西全部删除保存,然后再执行安装命令即可(看别人的博客都是添加各种镜像去配置,我也试过了不行,删除却ok了)
这篇关于ubuntu16.04系统pytorch安装的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-06-17zero-shot-learning-definition-examples-comparison
- 2024-06-06Package Easy(基于 NSIS 的打包exe安装包工具)使用方法-icode9专业技术文章分享
- 2024-06-06基于 casdoor 的 ELK 开源登录认证解决方案: elk-auth-casdoor-icode9专业技术文章分享
- 2024-05-29Elasticsearch慢查询日志配置
- 2024-05-29揭秘华为如此多成功项目的产品关键——Charter模板
- 2024-05-29海外IDC业务拓展的7大挑战
- 2024-05-29InLine Chat功能优化对标Github Copilot,CodeGeeX带来更高效、更直观的编程体验!
- 2024-05-29CodeGeeX 智能编程助手 6 项功能升级,在Visual Studio插件市场霸榜2周!
- 2024-05-29AutoMQ 生态集成 Apache Doris
- 2024-05-292024年IDC行业的深度挖掘:机遇、挑战与未来展望