最近一不小心将Linux环境变量里的pytorch误删除了,捣鼓一上午解决,在此记录下解决方案,避免大家像我一样踩坑。
一、Pytorch的安装(v1.0.0和v1.0.0之前的版本不一样安装方式)
1、V1.0.0及以后的版本
首先登陆pytorch官网https://pytorch.org/get-started/locally/,进入本地安装选项。
假设我的系统为Linux、语言为python、使用conda方式安装,cuda版本选择符合自己GPU驱动的版本(后面会列出对应关系)。
在命令行中输入
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
在安装时可能出现超时或者HTTP的相关错误。解决方案:添加国内的镜像源,这样问题就应该解决了。
# 配置国内源
#(清华源),方便安装Numpy等
conda config --add cha