ubuntu16.04+pytorch0.4.0+cuda9.0+Anaconda3-4.3.0 (python 3.6.0) 搭建深度学习实验环境爬坑指南！

 

踩坑1：我之前搭建了一版本pytorch0.4.0环境，可是因为我搭建的环境的python版本是Ubuntu系统自带的2.7.12，因为python版本过低,所以报错

huling@hmc32:~/cspm/c-s-p-m$ python train.py
Traceback (most recent call last):
  File "train.py", line 49, in <module>
    prev_val_loss = math.inf
AttributeError: 'module' object has no attribute 'inf'

出现该问题的原因是：我环境中使用的是python2，版本太低，于是换成python3以上的版本，可是我明明装了anaconda是3.5或者3.6的版本，可是仍旧报错。

于是我输入 python查看版本，显示是python2.7.12，代表环境中用的是默认的系统自带的2.7.12版本。尽管你已经装了python的其它版本，但是你并没有添加环境变量，所以并没有用到你装的其它版本。

huling@hmc32:~$ python
Python 2.7.12 (default, Oct  5 2020, 13:56:01)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.

然后我再次确认系统中自己成功装了哪些版本的python。使用命令：

 ls /usr/bin/python*

可以看出有python2.7、python3.5、python3.6

huling@hmc32:~$ ls /usr/bin/python*
/usr/bin/python            /usr/bin/python2.7-coverage  /usr/bin/python2-pbr       /usr/bin/python3.5m         /usr/bin/python3-config   /usr/bin/python-coverage
/usr/bin/python2           /usr/bin/python2-config      /usr/bin/python3           /usr/bin/python3.5m-config  /usr/bin/python3m
/usr/bin/python2.7         /usr/bin/python2-coverage    /usr/bin/python3.5         /usr/bin/python3.6          /usr/bin/python3m-config
/usr/bin/python2.7-config  /usr/bin/python2-jsonschema  /usr/bin/python3.5-config  /usr/bin/python3.6m         /usr/bin/python-config

再次查看系统默认版本，确认是ubuntu系统自带的2.7.12

huling@hmc32:~$ python --version
Python 2.7.12

修改系统默认的python版本，只需要在其home目录下创建一个alias。打开该用户的~/.bashrc文件：输入以下：

vim ~/.bashrc

在~/.bashrc文件 添加如下（python3.5或者你想添加的python3.6或者其它的版本呢）

alias python='/usr/bin/python3.5'

修改完之后，重新登录或者重新加载.bashrc文件，使操作生效：

$ . ~/.bashrc  

检查当前的版本

huling@hmc32:~$ python --version
Python 3.6.5

踩坑2：提示成功安装pytorch0.4.0，但是在python环境下，进行import torch报错，报出没有组件名字‘torch’，如下所示：

(python36) huling@hmc32:~$ python
Python 3.6.5 (default, May  3 2018, 10:08:28)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torch'

查看了很多，但是就是不奏效。于是突然发现自己进入的python的环境不是anaconda里面的。因为你安装的pytorch0.4.0是安装在anaconda环境中的，所以你需要使用anaconda中的python。

这写的是默版本的python，正确的是

这才是正确的。表明使用的是anaconda中的python。

修改默认python版本为anaconda中python版本

输入以下命令（其中path为anaconda的默认安装路径）。记得将用户名改为自己的用户名。（/home/huling/指的是huling改为自己的）

echo 'export PATH="/home/huling/anaconda3/bin:$PATH"' >> ~/.bashrc

重启个性化设置。

source ~/.bashrc