pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
可以直接在jupyter notebook使用安装命令,不知是什么原因用conda命令会卡死,pip3则出错,暂时在只能用pip命令,可以离线安装conda install --offline -n env_name XXXX…tar.bz2,env_name XXXX…tar.bz2为文件名,必须在默认目录下(网址可以在在线安装后获得)
pytorch安装包网址https://download.pytorch.org/whl/torch_stable.html
cuda10.2安装包https://files.pythonhosted.org/packages/a4/54/81b1c3c574a1ffde54b0c82ed2a37d81395709cdd5f50e59970aeed5d95e/torch-1.10.2-cp36-cp36m-manylinux1_x86_64.whl
conda安装包及cuda安装
最新推荐文章于 2024-05-23 16:06:06 发布