本文介绍了如何在Linux系统中通过创建虚拟环境来安装PyTorch，以避免与其他库冲突。首先，使用命令创建一个名为pytorch、基于Python 3.7的虚拟环境，然后激活该环境。接着，参照PyTorch官网的指导，输入conda安装命令conda install pytorch torchvision cudatoolkit=10.1完成安装。最后，验证PyTorch是否已成功安装。
创建虚拟环境
为pytorch创建一个虚拟环境，这是推荐的做法，这样在创建其他环境如tensorflow时，就不会互相冲突。在终端输入：
conda create -n pytorch python=3.6 
这就创建了一个名为pytorch，python版本为3.7的虚拟环境。于是我们可以把Pytorch安装在这个环境下面。首先我们激活这个环境：source activate pytorch
source activate pytorch激活后，会看到前面会显示（pytorch）【这就进入了pytorch的环境：】
最后，在Pytorch官网，根据自己的机器配置选择对应的安装命令。
即输入conda install pytorch torchvision cudatoolkit=10.1 就安装好了
然后可以检测下是否成功安装了pytorch
扫一扫
1796
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
