1.创建虚拟环境
conda create -n pytorch python==3.7 |
注:-n是name的意思,后面指定虚拟环境的版本为3.7
2.进入环境
conda activate pytorch |
注:conda可以省略
3.安装pytorch
打开官网:https://pytorch.org/
选择相应的版本,如果你的电脑没有英伟达显卡,cuda选择none
注意
4.配置conda源
找到此路径下的隐藏文件.condarc
编辑此文件
以下提供复制文件:
show_channel_urls: true ssl_verify: true channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge |
5.输入命令
6.检验安装
7.win10装cuda
网址:https://developer.nvidia.com/cuda-10.1-download-archive-base
选择相应版本进行安装我装的是cuda10.1,和我的虚拟环境对应
8.显卡驱动
若版本不符合去官网下载:
地址:https://www.nvidia.cn/geforce/drivers/
或者:https://www.nvidia.cn/Download/index.aspx?lang=cn