0
点赞
收藏
分享

微信扫一扫

安装Pytorch

有点d伤 2022-03-19 阅读 163
pytorch

安装Pytorch

安装anaconda

清华源https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
Anaconda3-5.2.0-Windows-x86_64.exe对应Python3.6
下载后点击安装即可(不要add to PATH)

配置环境

打开Anaconda Prompt
conda create -n pytorch python=3.6配置环境
安装完激活 conda activate pytorch
关闭 conda deactivate
查看当前环境工具包 pip list

安装pytorch

nvidia控制面板-系统信息-显示-细节 查看是否支持cuda以及驱动版本号 ptyorch.org查看对应安装命令
在pytorch环境中执行

Windows

cuda: conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
cpu: conda install pytorch torchvision torchaudio cpuonly -c pytorch

举报

相关推荐

0 条评论