1.正确安装anaconda

1)下载对应Anaconda版本,以Anaconda3-5.3.1-Linux-x86_64.sh为例。Anaconda下载连接
2)命令安装

bash Anaconda3-5.3.1-Linux-x86_64.sh

3)接下来根据提示信息选择yes or no
4)最后刷新一下,即安装成功

2.创建环境

此部分必须保证第一步已正确安装Anaconda。

conda create -n pytorch python=3.7 %创建一个名字为pytorch的环境,并给她安装3.7版本的python。

3.官网不稳定,从清华源下载安装

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

4. 安装对应版本的torchvision和cudatoolkit

CUDA 10.1
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 

5.安装成功确认

conda info -e #查看安装的环境

在这里插入图片描述

6.友情提示

其它版本,请参照官网conda安装命令,并把安装命令最后的 -c pytorch 删除,否则是从官网下载,网速特慢。

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐