概述
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所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复