0
点赞
收藏
分享

微信扫一扫

torch安装

乱世小白 2022-04-13 阅读 217
python

第n次安装了,每次都得结合很多篇才能安装成功,这次直接自己写一个。

一、安装anaconda

这个自己搜吧,挺简单的。

二、安装pytorch

进入anaconda prompt

创建一个虚拟环境(就不用再安装python了,创建之后直接就是用的3.8版本的python)

conda create -n env_name python=3.8

进入虚拟环境

conda create env_name

使用清华镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channels_urls yes

安装pytorch(官网)

  • 不要用pip并去掉-c python的方法(会默认下载cpu);
  • 如果电脑没有gpu那就不要选cuda11.3,选cpu 
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

三、验证

没报错,而且是True(电脑没有gpu的小伙伴就是false啦) 

举报

相关推荐

0 条评论