Linux安装Pytorch

时间:2022-01-14
本文章向大家介绍Linux安装Pytorch,主要包括Linux安装Pytorch使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

①安装Anaconda,Anaconda指的是一个开源的Python发行版本,其包含了conda、Python等180多个科学包及其依赖项。首先在Anaconda官网下载Linux版本的Anaconda,并通过MobaXterm或者XFtp传入服务器中。之后通过命令行进行安装。

sh Anaconda3-2021.11-Linux-x86_64.sh

4.7 安装Anaconda

②重新启动MobaXterm或Xshell后,进入base环境。添加清华镜像源以便后续安装pytorch。

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes

4.8 添加镜像源

③创建Pytorch环境,激活3dunet环境并安装pytorch以及代码所需要的库和模块(Simpleitk、Scipy、TensorBoard等)。

conda create -n Pytorch python=3.7
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2
conda install gpustat

4.9 创建环境

4.10 安装Pytorch

原文地址:https://www.cnblogs.com/HaloZoey/p/15800642.html