安装GPU版本
这里以Linux系统CUDA11.3以及python3.8为例,我们打开网址:
https://download.pytorch.org/whl/cu113/torch_stable.html
检索下torch-1.10.0然后下载下面用红色框框住的文件:
接着检索下torchvision-0.11.1然后下载下面用红色框框住的文件:
我们在这里可以找到我们需要的torch-1.10.0+cu113-cp38-cp38-linux_x86_64.whl以及torchvision-0.11.1+cu113-cp38-cp38-linux_x86_64.whl两个文件即可。注意,cu113代表CUDA11.3,cp38表示python3.8的编译环境,linux_x86_64表示x86的平台64位操作系统。下载完成后,我们将这两个文件传入你的离线主机(服务器)中。接着进入刚刚用conda创建好的虚拟环境后依次安装whl包:
cd 进入你的本人下载这两个包的位置
pip install torch-1.10.0+cu113-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.11.1+cu113-cp38-cp38-linux_x86_64.whl
出现
WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000002AF2B7CFCC8>, 'Connection to pypi.org timed out. (connect timeout=15)')': /simple/numpy/
将网络换成你的手机个人热点即可