-
首先在官网下载安装好Anaconda,目前pytorch支持python3.5-3.7,所以应该下载所需版本
-
安装CUDA传送门和CUDNN传送门,注意一定要和目前的pytorch支持的版本匹配,比如CUDA10.1,cuDNNv7.6.5。具体安装过程,参考此处传送门
-
在Anaconda中创建虚拟环境,在开始菜单中打开 Anaconda Prompt,然后在命令行输入
conda create -n pytorch python=3.x
以创建虚拟环境 -
到pytorch官网点击此处获取安装链接。下面是加了清华镜像源的链接,这样下载更稳定,也更快;
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.4.0 torchvision ==0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
-
然后激活刚才创建的虚拟环境,使用
conda activate pytorch
,将上述的下载链接输入到命令行,等待安装完成 -
最后使用pip,安装各种需要的包即可,注意是在虚拟环境下使用pip
pytorch学习笔记一:简述在Anaconda下安装pytorch1.4的步骤
关注
打赏