目录
安装pytorch
官方网址:
Start Locally | PyTorchhttps://pytorch.org/get-started/locally/
官网的conda安装方式,没有指定各种组件的版本,因为好多算法是规定了pytorch版本,这种可读性不高的命令行安装方式我没有采取。
我采取的是找到之前的版本,但是要求他比较新,因为算法在迭代,pytorch版本低,之后还得重新配置虚拟环境。
选一个合适的版本,根据我之前的学习现在比较新的算法都需要pytorch1.10以上了,我选择1.11.0。
注意!!!pytorch1.6.0在练习pytorch时会出现torch.nn没有代码提示!!!避免安装pytorch1.6.0.
以下是详细的配置环境过程
采用conda安装方式,名称(-n)为pytorch,python版本为3.7。
conda create -n pytorch python=3.7
进入虚拟环境。
conda activate pytorch
采用pip安装,第一步就安装了pip安装工具
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
查看cuda版本,更新的最新肯定是没问题的,向下兼容。
nvidia-smi
注意cuda版本是向下兼容的,把显卡驱动升级到最新即可。
检查pytorch安装是否成功
以下代码全部运行成功,才算安装成功。
python
import torch
import torchvision
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
exit()
总结
1.注意别安pytorch1.6.0版本,torch.nn没有代码提示,不利于学习。
2.cuda版本不用严格与系统的cuda版本一致,但是系统的cuda版本要比pytorch要求的cuda版本高。