pytorch 安装教程
创建虚拟环境
conda create -n pytorch_gpu python=3.8
查看python 所有虚拟环境
conda info -e
激活环境
source activate
退出环境
source deactivate
激活指定环境
conda activate ptorch_gpu
安装 pytorch
# CUDA 11.1
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
查看pytorch 版本exit
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
安装jupyter
conda install jupyter notebook
生成配置文件
jupyter notebook --generate-config
打开ipython
from notebook.auth import passwd
passwd()
修改配置文件
vim ~/.jupyter/jupyter_notebook_config.py
杀死gup
kill -9 PID
服务器创造jupyter连接
jupyter notebook --no-browser --port=9999 --allow-root