赞
踩
不需要按网上教程去官网下载,也不需要手动在conda环境内安装。直接按PyTorch官网命令安装PyTorch即可,cuda和cudnn会自动按你选择的版本下载。
大家应该都会
- conda create -n 你的环境名字 python=你想要的python版本
- 如:
- conda create -n pytorch python=3.6
conda activate 你的环境名字
官网链接:Previous PyTorch Versions | PyTorch
如果不需要特定版本,目前比较稳定的版本可选pytorch 1.8.0 +CUDA 10.2。激活环境后直接运行下面代码即可:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
其他版本按官网指示来,需要改动几个数字
重点:不要安装最新版和带有".1"的版本,比如不要装 1.8.1(大坑!)。因为后面的Pytorch geometric (Pyg)兼容性不够好,会出各种问题。
更新:2022/4/26,上述命令不能用了,建议大家去官网使用最新版的指令
- python
- import torch
- print(torch.__version__)
- print(torch.cuda.is_available())
返回你的PyTorch版本和Ture既是安装成功
按CTRL+D退出python,如果上一步选择的PyTorch1.8.0版本(或1.8.0以上),此时直接运行下面代码即可,亲测1.8.0没问题,安装完成
conda install pyg -c pyg -c conda-forge
其他版本按官网指示Installation — pytorch_geometric 2.0.4 documentation
找到对应的pip代码运行即可
如果之前安装出现问题,建议新建一个环境,不在老环境里修改。新环境第一个就装PyTorch,会自动装需要的其他包。
用conda命令安装时它会检测你环境里缺少的包,然后自动给你一起安装。这很方便,但是有时会装一些不正确的包,导致一些冲突。所以一定要检查它要给你装哪些包,确认没问题在确认安装。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。