centos7安装pytorch0.4.0 (别的版本也适用)
- 创建一个新的conda环境
创建python版本为X.X、名字为your_env_name的虚拟环境。your_env_name文件可以在Anaconda安装目录envs文件下找到。
conda create -n your_env_name python=X.X(2.7、3.6等)
- 查看所有conda环境
conda env list
- 查看新建的conda环境中所安装过的库
pip list
- 去pytorch官网查找对应的pytorch版本,进行安装
官网链接:https://pytorch.org/get-started/previous-versions/
在网页下面找到自己想要安装的版本,复制命令进行安装。选择版本的时候注意自己的cuda版本,我的centos7中的cuda是10.1,但是使用conda install pytorch=0.4.1 cuda90 -c pytorch命令安装的pytorch没有问题,可以使用。
查看cuda版本命令
cat /usr/local/cuda/version.txt
安装torchvision
pip install torchvision==0.2.1
- over