要安装特定版本的PyTorch(pytorch==1.0.1),可以按照以下步骤进行操作:
pip install torch==1.0.1
这将使用pip包管理器从Python软件包索引中安装指定版本的PyTorch。
import torch
print(torch.__version__)
如果输出为1.0.1,则表示已成功安装了特定版本的PyTorch。
请注意,如果您使用的是Conda环境,可以使用类似的命令来安装特定版本的PyTorch:
conda install pytorch=1.0.1 torchvision cudatoolkit=10.0 -c pytorch
这将使用conda包管理器安装指定版本的PyTorch。
希望对您有所帮助!
下一篇:安装特定版本的sklearn