anaconda安装及pytroch-gpu环境配置

发布时间：2022-08-19 12:41

该文为自己使用的笔记

anaconda下载地址:
https://www.anaconda.com/products/distribution
下载后直接运行即可
安装后换源,据传清华源过期了并有图片(注意是19年的不是现在的),不过不知道现在是否已经可用,鉴于之前用清华源有过不太好的体验,所以我改用了北京外国语大学的镜像源
打开conda prompt,可以直接操作也可以新建一个环境
如果已经安装了pytorch-cpu版本,可以先删除
删除方法:
如果是用conda安装的,输入:

conda uninstall pytorch

conda uninstall libtorch

如果是用pip安装的,输入:

pip uninstall torch

不记得用什么装的就都试试

然后选择并安装cuda:
https://developer.nvidia.com/cuda-downloads
之后安装pytorch(注意选择pip而不是conda,不要选择cpu版本):
https://pytorch.org/get-started/locally/
理论上在这里选择conda或pip结果是一样的,但是因为前面换了北外的源,这里用conda下载会下载为cpu版本
换成pip下载则对网络有要求,网络不好容易time out,自己取舍

然后打开conda prompt测试安装是否成功:

python
import torch
torch.cuda.is_available()

返回True就是GPU版本了

anaconda安装及pytroch-gpu环境配置

相关推荐