目录

安装pytorch

检查pytorch安装是否成功

总结

安装pytorch

官方网址：

Start Locally | PyTorch/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 /whl/cu113查看cuda版本，更新的最新肯定是没问题的，向下兼容。nvidia-smi

注意cuda版本是向下兼容的，把显卡驱动升级到最新即可。

检查pytorch安装是否成功

以下代码全部运行成功，才算安装成功。

pythonimport torchimport torchvisionprint(torch.__version__)print(torch.version.cuda)print(torch.cuda.is_available())exit()

总结

1.注意别安pytorch1.6.0版本，torch.nn没有代码提示，不利于学习。

2.cuda版本不用严格与系统的cuda版本一致，但是系统的cuda版本要比pytorch要求的cuda版本高。

如果觉得《1.pytorch学习:安装pytorch》对你有帮助，请点赞、收藏，并留下你的观点哦！