1.安装Anaconda
清华镜像源Index of /anaconda/archive/ | 北京外国语大学开源软件镜像站,致力于为国内和校内用户提供高质量的开源软件镜像、Linux 镜像源服务,帮助用户更方便地获取开源软件。本镜像站由清华大学 TUNA 协会负责运行维护。https://mirrors.bfsu.edu.cn/anaconda/archive/使用清华镜像源下载安装包,也可以去官方网站下载,(推荐使用镜像源下载,下载速度快)
双击此程序,安装anaconda。
2.环境配置
1.安装完成后,点击开始菜单,进入Anaconda Prompt命令行
2.在命令窗口添加下面命令
conda create --n pytorch python=3.6#创建新的环境
pytorch为环境名字,python=**可以甚至需要的版本。
出现上面的选择,输入Y.。然后等待库的安装。
3.根据提示激活环境。
activate pytorch
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/
# 添加pytorch镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
5.安装pytorch
进入pytorch官网,选择适合自己版本的conda(点击橘色字体)。
因为我自己电脑版本为11.2,所以我下载的11.2版本。win+r输入cmd打开命令窗口,输入”nvidia-smi“查看自己的版本。
conda install pytorch torchvision cudatoolkit=11.2
主要是因为
-c pytorch
表示指定使用pytorch channel,这个服务器位于国外,下载速度很慢,所以去掉了-c pytorch。
6.验证
1.python #会出现python版本,进入python内
2.import torch
3.torch.cuda.is_available()
依次输入上述命令,若最终返回True ,表示安装成功。
输入exit()退出即可。