欢迎您访问 最编程 本站为您分享编程语言代码,编程技术文章!
您现在的位置是: 首页

如何在conda虚拟环境中安装PyTorch和D2L包?

最编程 2024-01-19 10:32:32
...

conda虚拟环境安装pytorch

1、首先,conda终端添加清华镜像源,可以加快安装速度。

2、确认电脑匹配的CUDA型号,(例如,9.2)

3、新建一个虚拟环境,在终端运行 conda create -n XXX python=3.8

4、激活该虚拟环境,conda activate XXX

5、pytorch官网,找与CUDA对应版本的安装语句

 

6、在虚拟环境下,运行该语句,因为添加了镜像源通道,删除后面的“-c pytorch”。

7、等待一段时间,就会安装结束。

8、确认pytorch是否安装成功,在终端输入,

 

 

根据李沐 pytorch讲解,需要安装一个d2l包,安装过程如下

1、下载whl :https://www.cnpython.com/pypi/d2l/dl-d2l-0.15.1-py3-none-any.whl

 

 

 2、切换到虚拟环境(d2l),将下载的文件保存到 该虚拟环境下 Lib\site-packages,然后在终端运行

 

 3、查看该虚拟环境下已经安装的包,conda list,检查是否成功安装