在安装PyTorch之前要先安装好CUDA、cudNN、以及anaconda,还有就是编译器pycharm,然后你才能开始安装PyTorch。
目录
1、创建虚拟环境。
2、去PyTorch官网。
3、发现警告。
4、解决办法。
5、pip list 查看python中安装了哪些第三方库。
6、测试是否安装成功PyTorch。
1、创建虚拟环境。
创建一个PyTorch环境:conda create -n pytorch15 python=3.7激活环境:conda activate pytorch15查看虚拟环境:conda-env list
2、去PyTorch官网。
找到之前的版本连接: Previous PyTorch Versions | PyTorch 找到PyTorch1.5.0版本,以及对应的cuda10.1然后打开cmd,输入下面的代码:
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
3、发现警告。
如下图黄色的地方是一个警告,表示当前使用的pip版本过低。
4、解决办法。
输入 python -m pip install -U --force-reinstall pip,就会显示安装新版本成功。
5、pip list 查看python中安装了哪些第三方库。
6、测试是否安装成功PyTorch。
(1)打开cmd:输入 python,进入python环境;
(2)输入:import torch;
(3)输入:print(torch.__version__);
(4)输入:print(torch.cuda.is_available()),查询cuda是否可用,返回True表示没问题。
以上就是关于PyTorch的安装过程,希望对大家有帮助,我们一起学习!



