1.在服务器上面创建cuda10容器
2.安装anaconda
3.然后安装pytorch:
conda install pytorch torchvision cudatoolkit=10.0
4.创建虚拟环境:
conda create -n pytorch27 python=2.7
然后进入到创建的虚拟环境:
conda activate pytorch27
然后在虚拟环境中安装pytorch:
conda install pytorch torchvision=0.5.0 cudatoolkit=10.0
tips:这里是cudatoolkit=10.0
为什么要是torchvision=0.5.0,因为在cudatoolkit=10.0的情况下,torchvision默认安装的是0.1.8,pytorch默认安装的是0.1.12.而我们需要的torchvision版本是0.5.0,pytorch版本是1.4.0,在torchvision和pytorch版本都符合的时候,我们才能够安装下一步的faiss.
5.faiss的版本低于1.5.1的时候,会提示:
AttributeError: 'module' object has no attribute 'StandardGpuResources'
,点着是解决方案,
原因是因为`faiss的安装取决于cudatoolkit的版本,在cudatoolkit=10.1,python2.7的环境下faiss安装只能安装faiss 版本1.4.1的,所以在第四步中,将cudatoolkit=10.0并且torchvision=0.5.0.
然后在安装faiss版本为1.6.3的之后,跑程序抛出:
AttributeError: 'Clustering' object has no attribute 'obj'
这是由于faiss版本太高的原因,所有我就安装的faiss版本为1.5.3.
安装faiss:
conda install faiss-gpu=1.5.3
然后安装scikit-learn,Scipy,matplotlib:
conda install scikit-learn Scipy matplotlib