简单安装pytorch ubuntu conda

最新推荐文章于 2024-06-15 20:08:29 发布

哀酱

最新推荐文章于 2024-06-15 20:08:29 发布

阅读量461

点赞数

分类专栏：杂七杂八

本文链接：https://blog.csdn.net/u010821666/article/details/89917855

版权

杂七杂八专栏收录该内容

26 篇文章 0 订阅

订阅专栏

环境
Ubuntu16.04 cuda 9.0 python3.7 conda3
装torch简单的方法

#加快conda 下载速度
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --set show_channel_urls yes
#建虚拟环境
conda create -n py37_torch python=3.7
conda activate py37_torch

#根据https://pytorch.org/get-started/locally/官网指导安装torch
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
#但是……下载还是很慢
#果断用pip
pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

#最简单安装方法，完成！