1. 创建和激活新的环境
conda create -n pytorch python=3.6 conda activate pytorch
2. 安装pytorch前的准备
2.1 查看设备型号
安装pytorch之前需要安装cuda,调用gpu加速,所以需要知道gpu的型号
1、查看显卡型号
方法一:打开设备管理器->显示配适器,查询到为NVIDIA GeForce GTX 1050
方法二:打开任务管理器->性能
2、查看驱动版本
打开NVIDIA控制面板->帮助->系统信息
查看cuda适配版本
根据驱动版本选择对应的cuda版本
2.3 安装cuda
cuda官网链接,翻进去傻瓜式安装
老版本的cuda链接:链接
输入nvcc -V
查看是否安装成功
3. 安装pytorch
4、根据步骤1、2查询到的信息,进入pytorch官网,https://pytorch.org/,选择对应版本,复制指令,进入anaconda对应环境下输入指令安装pytorch
先换源,下载的快
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
后复制下面指令,注意去掉-c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2
报错:
看别人博客链接,说是因为anaconda不是默认装在C盘才报错
解决方法步骤:找到anaconda文件夹右击打开属性->安全->修改为system
再一次运行安装指令,之后便安装成功。
4. 测试
打开pycharm或者就在终端上测试:
import torch torch.cuda.is_available()
返回为True