一、写到前面
确保自己已经安装好了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下。
参考: