安装PyTorch
登入PyTorch官网:PyTorch
选择操作系统,cuda模式,复制安装命令。
Anaconda Prompt激活pytorch环境,执行命令。慢慢等安装吧。
如果你发现安装很慢的话,建议给anaconda添加国内源。而且我不建议是清华源,因为太卡了。这里推荐一下北外的源。具体操作如下:
打开用户目录下的.condarc文件,更新成为如下内容。
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
- https://mirrors.bfsu.edu.cn/anaconda/pkgs/r
- https://mirrors.bfsu.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.bfsu.edu.cn/anaconda/cloud
msys2: https://mirrors.bfsu.edu.cn/anaconda/cloud
bioconda: https://mirrors.bfsu.edu.cn/anaconda/cloud
menpo: https://mirrors.bfsu.edu.cn/anaconda/cloud
pytorch: https://mirrors.bfsu.edu.cn/anaconda/cloud
simpleitk: https://mirrors.bfsu.edu.cn/anaconda/cloud
命令行中执行命令清理缓存。
conda clean -i
安装完成之后,查看pytorch版本。
conda list
总结
基本环境已经搭好了,找个模型玩玩吧。