我是靠谱客的博主 殷勤世界,最近开发中收集的这篇文章主要介绍Ubuntu 下 conda 加速安装pytorch,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

1.确定自己的CUDA版本

cat /usr/local/cuda/version.txt

2.建立虚拟环境

conda create -n pytorch

开启、退出、删除命令:

source activate pytorch
source deactivate
conda remove -n pytorch --all

3.安装pytorch

https://pytorch.org/ 在官网主页根据你的系统和CUDA,python版本,选择conda安装方式。

使用官方指定的方法下载太慢,所以:

  • 进入anaconda 的清华镜像帮助:网站为:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
  • 复制以下命令,粘贴回车
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  • 输入官方提供的安装pytorch的命令(去掉末尾的-c pytorch)
    示例:
 conda install pytorch torchvision cudatoolkit=10.0

4.测试

根据提示安装好了之后可以python然后import torch尝试是否安装成功了。

最后

以上就是殷勤世界为你收集整理的Ubuntu 下 conda 加速安装pytorch的全部内容,希望文章能够帮你解决Ubuntu 下 conda 加速安装pytorch所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(55)

评论列表共有 0 条评论

立即
投稿
返回
顶部