小白pytorch安装
完全的代码小白pytorch安装记录
我是真的没什么经验,这里面的知识可能不够权威,仅供参考。
Anaconda3安装
官网下载安装即可
- 打开anaconda的prompt;
- 查看自己的电脑是否支持gpu版本,我的电脑支持,并且是NVIDIA960M;
- conda create -n pytorch python=3.5;(我使用的Python3.5)
- conda activate pytorch;
Pytorch安装
- 打开pytorch官网选择适合自己的版本,这里往往会很慢,因此采用清华镜像
- 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 - conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- conda install pytorch torchvision cudatoolkit=10.1
上述依次复制粘贴即可
这里需要自己的GPU驱动在400+,否则可能出现错误,我的电脑就是需要去NVIDIA升级自己的驱动,升级完就OK了。
我这里一开始倔强用官网方法,一下午都没搞定,用清华镜像几分钟搞定。
Pycharm安装
1.官网下载安装,我选择免费的社区版。至此,经测试我的安装完成且成功。