一、写到前面

确保自己已经安装好了Anaconda,确保自己要安装的cuda runtime版本是多少

比如我这里Anaconda安装包版本是:Anaconda3-2024.02-1-Windows-x86_64.exe

我结合自己电脑的硬件显卡名字是RTX 4060,算力是8.9,然后结合显卡驱动cuda driver版本

最终确定自己电脑要安装的cuda runtime版本是12.1

然后在Anaconda里面创建自己的虚拟环境,并要求Python版本是3.8

然后记得激活自己创建的虚拟环境。

接下来就是在自己创建的虚拟环境里面安装PyTorch

二、开始安装PyTorch

接着打开PyTorch官网

安装最新版本PyTorch 2.2.2

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

 分析:这里如果使用conda命令安装发现总是报错如下

Downloading and Extracting Packages:


CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/nvidia/win-64/libcufft-dev-11.0.2.4-0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/nvidia/win-64/libcufft-dev-11.0.2.4-0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

于是调整思路,干脆使用pip命令方式安装PyTorch也可以

于是就是漫长的等待即可。

接下来想看一下这个pip3 install的包安装到了哪里?

pip show packagename #这样会显示安装包的路径

一般来说安装的包会在python路径下的lib/site_packages。 如果是在自己创建的虚拟环境下通过pip安装的包,会在虚拟环境下的lib/site_packages下。

 

参考:

查看pip安装包的位置以及修改默认安装路径_pip 位置-CSDN博客

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐