Anaconda下安装Pytorch-GPU版
一、Anaconda创建、激活、退出、删除虚拟环境
1.1 创建虚拟环境
打开终端,输入以下指令创建环境:
conda create -n tf python=3.6
conda 是指调用 conda 包,create 是创建的意思,-n 是指后面的名字是环境的名字, tf是环境的名字,python=3.6 是指python3.6 版本。
1.2激活虚拟环境
使用如下命令即可激活创建的虚拟环境
Linux: source activate tf(虚拟环境名称)
Windows: activate tf(虚拟环境名称)
使用python -V检查当前python版本
1.3退出虚拟环境
使用如下命令即可退出创建的虚拟环境
Linux: source deactivate tf
Windows:deactivate tf
,也可以使用activate root
切回root环境
1.4删除虚拟环境
删除环境:
使用命令conda remove -n tf --all
, 即可删除。
删除虚拟环境中的包:
使用命令conda remove --name $your_env_name $package_name(包名)
即可。
1.5管理包
需要安装时,直接输入 conda install包名。
例如,如果需要安装numpy,输入 conda install numpy
例如: conda install numpy scipy pandas
会同时安装。也可以指定版本号,例如 conda install numpy=1.10
conda会自动安装依赖。例如,scipy依赖于numpy。如果你只安装scipy(conda install scipy),Conda会自动安装numpy如果之前没安装过。
如果需要删除,直接使用 conda remove包名,更新包 conda update package_name。
如果需要更新环境中所有库, conda update–all。如果需要列出已安装软件, conda list。
如果你不知道确切的包名,可以用 conda search,尝试用 conda search beautifulsoup。
conda install 包 可能会升级其他的包
临时使用清华镜像
pip install -i https://pypi.mirrors.ustc.edu.cn/simple some-package
二、matplotlib中文显示问题
2.1中文字体
修改配置文件matplotlibrc(注意虚拟环境的目录)
在matplotlib的安装路径:python3.6\site-packages\matplotlib\mpl-data\matplotlibrc,文件中有如下内容:
#font.family : sans-serif
#font.sans-serif : DejaVu Sans, Bitstream Vera Sans,, Helvetica, Avant Garde, sans-serif
然后将开头的两个注释符删掉。
2.2 matplotlib字体库
将中文字体:simhei.ttf复制到python3.6\site-packages\matplotlib\mpl-data\fonts\ttf文件夹,并在font.sans-serif冒号后面加上SimHei。然后删除~/.cache/matplotlib的缓冲目录
2.3 负号显示
#axes.unicode minus : True
为了能够使坐标轴上的负号正常显示,同样将开头的注释符#去掉,然后将True改为False。
三、tensorflow-gpu安装
推荐安装(注意可能会修改python版本),会自动安装cuda,cudnn:
conda install tensorflow-gpu
四、pytotch-gpu安装
添加清华镜像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
安装命令:
conda install pytorch torchvision cudatoolkit=10.1
五、导出环境
导出环境到yml文件,文件名为 python36.yml
conda env export > python36.yml
将yml文件复制到B机器中,执行以下命令导入
conda env create -f python36.yml