博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Ubuntu16.04下基于Conda安装PyTorch
阅读量:4279 次
发布时间:2019-05-27

本文共 274 字,大约阅读时间需要 1 分钟。

Ubuntu16.04下基于Conda安装PyTorch

安装环境

本文的安装过程适用于在Ubuntu下已经安装好GPU和conda的计算机。

翻墙

不翻墙装软件会报错。

命令行安装

在这里插入图片描述

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

安装验证

$ jupyter notebook

在.ipynb中输入:

import torchimport torchvisionprint(torch.cuda.is_avaiable())

显示输出:

True

参考资料

[1]

[2]

转载地址:http://vlzgi.baihongyu.com/

你可能感兴趣的文章
leetcode-14:求字符串数组最长公共前缀
查看>>
TCP三次握手,四次握手过程原理
查看>>
红黑树和AVL树(平衡二叉树)区别
查看>>
lightGBM使用教程
查看>>
EM算法推导
查看>>
xgboost原理与推导
查看>>
python 中的map()函数
查看>>
python中 lambda 来创建匿名函数
查看>>
Python是如何进行内存管理-内存池机制Pymalloc
查看>>
树、森林和二叉树的转换
查看>>
Hadoop和Spark的区别
查看>>
词向量—Word2Vec入门及Gensim实践
查看>>
卷积神经网络CNN 中用1*1 卷积过滤器的作用及优势
查看>>
深度学习: 卷积核尺寸size为什么是 奇数
查看>>
机器学习比赛—杀入Kaggle Top 1%
查看>>
数据如何归一化 类型
查看>>
Summarization 文本摘要进展
查看>>
kmeans聚类算法及复杂度
查看>>
四大机器学习降维算法:PCA、LDA、LLE、Laplacian Eigenmaps
查看>>
批标准化(Batch Normalization)
查看>>