安装时会同时安装CUDA Toolkit以及PyTorch,所以不需要下载额外的CUDA包。
CUDA版本可在cmd中输入:nvidia-smi
进行查询。
Anaconda的安装方法见我另一篇博客
Win10+GTX1660Ti+Anaconda3+Python3.7+TensorFlow-gpu1.14.0+Pycharm2020.3.2环境搭建记录
以下是安装方法。
打开Anaconda Prompt的第一件事:换源!不换源真的慢的要死。
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
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 --set show_channel_urls yes
直接复制就行。有警告不要怕,一般是说你的url列表里已经有这个镜像源的链接了。
第二步,更新anaconda-navigator。后面安装pytorch时,如果要用conda指令安装,那么这一步必不可少。
conda update anaconda-navigator
第三步,创建pytorch环境。
conda create --name pytorch python=3.7
name后面是你的环境名,我直接设置的“pytorch”,python后面是你指定的python版本。然后就是下载一堆东西。
创建完成之后激活环境
activate pytorch
然后在pytorch官网查询下载指令
因为我安装的时候还不知道要更新anaconda-navigator,conda指令就报错,所以我直接用pip了。
下载还挺快。安装之后,重新启动Anaconda Prompt,进入pytorch环境,python输入
import torch
不报错就代表安装成功。
然后再输入
torch.cuda.is_available()
显示True,则代表CUDA安装成功。