LINUX安装Anaconda和pytorch
-
wget +(https://repo.anaconda.com/archive/ +自己需要的版本
比如 wget https://repo.anaconda.com/archive/Anaconda3-2021.11-Linux-x86_64.sh
安装完成以后
输入 ls ,显示的是Anaconda3-2021.11-Linux-x86_64.sh
-
sh Anaconda3-2021.11-Linux-x86_64.sh
然后一直回车或者yes,anaconda安装好了
-
在anaconda中配置清华源
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
-
安装对应的pytorch或者TF,本人用的为pytorch,可在官网搜索
conda install pytorch torchvision torchaudio cudatoolkit=11.3
切记,用国内源不需要加 -c pytorch,这东西代表在官方下载
-
检测torch能否使用
import torch print(torch.__version__) print(torch.cuda.is_available()) #返回当前设备索引 torch.cuda.current_device() #返回GPU的数量 torch.cuda.device_count() #返回gpu名字,设备索引默认从0开始 torch.cuda.get_device_name(0)