pytorch 环境配置

发表于 | 分类于

Anaconda+pytorch

使用Anaconda

Anaconda 是 Python 的一个开源发行版本，主要面向科学计算,可以简单理解为，Anaconda 是一个预装了很多我们用的到或用不到的第三方库的 Python。而且相比于大家熟悉的 pip install 命令，Anaconda 中增加了 conda install 命令。

https://www.anaconda.com/products/individual # anaconda 下载地址

1
2
3
4
5
6
7
8
wget https://repo.anaconda.com/archive/Anaconda3-2020.02-Linux-x86_64.sh  # 可以去官网上找合适的版本
bash Anaconda*-Linux-x86_64.sh -b -p $(pwd)/anaconda3
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/  # 换源
conda config --set show_channel_urls yes  # 设置搜索时显示通道地址

conda config --set auto_activate_base false # 取消自动进入 conda 环境

conda activate base  # 激活

安装 pytorch

1
2
3
4
5
6
 ./conda create --prefix=/anaconda/anaconda3/myenv  --name my_first_env  python=3.8
conda activate my_first_env

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
# conda install pytorch torchvision cpuonly -c pytorch  #  cpu only 
# conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge # 根据 GPU 版本

pip+pytorch

验证

##

使用 jupyter

1
2
3
4
5
jupyter notebook --generate-config

vi /root/.jupyter/jupyter_notebook_config.py

nohup jupyter notebook --allow-root > jupyter.log 2>&1 &

写入的配置文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
from notebook.auth import passwd
passwd()

# 允许通过任意绑定服务器的ip访问
c.NotebookApp.ip = '*'
# 用于访问的端口
c.NotebookApp.port = 8527
# 不自动打开浏览器
c.NotebookApp.open_browser = False
# 设置登录密码
c.NotebookApp.password = 'sha1:14855cd59712:1cf1063d38e08cd2703a07a52b66714281676b6d'
# 设置默认目录
c.NotebookApp.notebook_dir = u'/root/'
c.NotebookApp.base_url = '/'

###
jupyter 还可以搭配 nginx， 切换主题，不展开说明

jupyter参考

卸载

rm -rf /anaconda/anaconda3

删除 .bashrc 中的相关字段