文章目录
一、安装Anaconda
需要说明的是,安装anaconda的对应python版本不影响我们在使用过程中创建其他python版本的环境,如我下载的是:python3.8版本的conda,那我在使用过程中依然可以通过指定python3.6版本的方式创建环境,conda会自动下载python3.6版本的所有基本包,你也可以自己另外安装其他所需要的任何库。
1.下载任意版本的conda安装脚本
2.安装
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sh Miniconda3-latest-Linux-x86_64.sh
之后长安回车和全输入yes即可。
我的安装路径设置为::
/home/pc/Softwares/anaconda/miniconda3
3.测试
source ~/.bashrc
python
如果出现,则表示安装成功:
4.换源
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/pytorch/
conda config --set show_channel_urls yes
卸载(如有需要)
安装pytorch
创建conda虚拟环境
conda create -n pytorch python=3.6.9
conda activate pytorch
卸载虚拟环境(如有需要)
conda deactivate # 退出虚拟环境
conda remove -n pytorch --all
查看cuda版本
nvcc -V
选择pytorch对应版本
安装
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
测试
进入python环境:
python
输入:
import torch
import torchvision
print(torch.cuda.is_available())
如果出现False:
1.检查安装的pytorch是否为GPU版本
果然后缀为cpu:
2.卸载cpu版本
conda uninstall pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
在网上发现pytorch1.4.0版本没有cuda10.2对应版本的whl,所以现在在下载时要指定版本:(官网版本查询处)
conda install pytorch=1.5 torchvision=0.6 cudatoolkit=10.2 -c pytorch
安装前检查:
测试:
参考文章: