Pytorch环境搭建

新建conda环境

conda create -n pytorch python=3.7

进入环境

conda activate pytorch

Pytorch环境安装

指定版本安装（比如v1.8.1）：
注意两个等号和版本号之间的连接没有空格，由于md格式导致书写有问题。

conda install pytorch == 1.8.1 torchvision == 0.9.1 torchaudio==0.8.1 cpuonly -c pytorch

如果需要安装其他版本，登录pytorch官网，选择和自己cuda版本对应的pytorch即可。
登不上pytorch官网，没法截图。
在powershell中，输入命令： nvidia-smi，查看自己cuda版本

安装成功后环境验证：

import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号

按照上面的步骤安装的结果：

到这里，pytorch就成功安装啦！
其他参考链接：
windows10上conda安装pytorch+transformers
pytorch官网被墙了，pytorch最新安装方法，可靠
安装和配置 PyTorch

jupyter notebook环境安装

安装命令：

conda install -y jupyter
conda install ipykernel

将conda环境添加到jupyter的内核中

python -m ipykernel install --name pytorch

如果安装中报错：ModuleNotFoundError: No module named ‘jupyter_server.contents‘
参照以下解决方法：解决方法

torchtext安装

参考链接