安装pytorch
·
一、环境隔离
创建一个pytorch环境
conda create -n pytorch python=3.10
常用的环境指令
conda info -e //查看已有环境
conda create -n env_name Python=3.x //创建新的 Python 环境
conda activate env_name //进入已有的 Python 环境
conda deactivate //退出当前的 Python 环境
conda remove -n env_name --all //删除 Python 环境
二、pytorch安装
1、进入pytorch官网(PyTorchPyTorchPyTorch)

2、按win+R后,输入cmd,在出现界面输入“nvidia-smi”,查询自己电脑显卡的 CUDA 版本(看CUDA Version)

3、在pytorch官网的Compute Platform选项中,选择低于自己显卡的 CUDA 版本。若没有更低的版本,点击Previous versions of PyTorch,查看以前的版本进行选择。
4、复制Run this Command中的内容
5、打开anaconda prompt,进入pytorch环境,粘贴复制内容,等待安装完成
三、检验安装
检验 Pytorch 是否安装成功,是否能成功调用显卡:
1、输入“Python”,并按回车,进入 Python 环境。
2、输入“import torch”,并按回车,若无报错,说明 Pytorch 安装成功。
3、输入“torch.cuda.is_available()”,并按回车,若显示“True”,说明可调用显卡。
四、在pytorch环境中安装jupyter notebook
进入pytorch环境安装包
conda install nb_conda
如果安装失败,运行以下代码
conda install nb_conda_kernels
更多推荐


所有评论(0)