{"id":960039,"date":"2024-12-27T03:46:08","date_gmt":"2024-12-26T19:46:08","guid":{"rendered":"https:\/\/docs.pingcode.com\/ask\/ask-ask\/960039.html"},"modified":"2024-12-27T03:46:10","modified_gmt":"2024-12-26T19:46:10","slug":"python%e4%b8%ad%e5%a6%82%e4%bd%95%e4%b8%8b%e8%bd%bdtorch","status":"publish","type":"post","link":"https:\/\/docs.pingcode.com\/ask\/960039.html","title":{"rendered":"python\u4e2d\u5982\u4f55\u4e0b\u8f7dtorch"},"content":{"rendered":"

\"python\u4e2d\u5982\u4f55\u4e0b\u8f7dtorch\"<\/p>\n

\u5728Python\u4e2d\u4e0b\u8f7dTorch\u7684\u65b9\u6cd5\u6709\u591a\u79cd<\/strong>\uff0c\u53ef\u4ee5\u901a\u8fc7pip\u547d\u4ee4\u3001conda\u547d\u4ee4\u6216\u8005\u76f4\u63a5\u4ece\u6e90\u7801\u7f16\u8bd1\u5b89\u88c5\u3002pip\u547d\u4ee4\u662f\u6700\u5e38\u7528\u7684\u65b9\u6cd5<\/strong>\uff0c\u56e0\u4e3a\u5b83\u7b80\u5355\u3001\u5feb\u901f\uff0c\u4e14\u9002\u7528\u4e8e\u5927\u591a\u6570\u7528\u6237\u3002\u4f7f\u7528pip\u547d\u4ee4\u5b89\u88c5\u65f6\uff0c\u9700\u8981\u786e\u4fdd\u73af\u5883\u4e2d\u5df2\u7ecf\u5b89\u88c5\u4e86Python\u548cpip<\/strong>\u3002\u5177\u4f53\u6b65\u9aa4\u5982\u4e0b\uff1a<\/p>\n<\/p>\n

\u9996\u5148\uff0c\u786e\u4fdd\u4f60\u5df2\u7ecf\u5b89\u88c5\u4e86Python\u548cpip\u3002\u5982\u679c\u6ca1\u6709\u5b89\u88c5\uff0c\u53ef\u4ee5\u4ecePython\u5b98\u65b9\u7f51\u7ad9\u4e0b\u8f7d\u5e76\u5b89\u88c5Python\uff0cpip\u901a\u5e38\u4f1a\u968fPython\u4e00\u8d77\u5b89\u88c5\u3002\u63a5\u4e0b\u6765\uff0c\u6253\u5f00\u547d\u4ee4\u884c\u6216\u7ec8\u7aef\uff0c\u8f93\u5165\u4ee5\u4e0b\u547d\u4ee4\u6765\u5b89\u88c5Torch\uff1a<\/p>\n<\/p>\n

pip install torch<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u5728\u67d0\u4e9b\u60c5\u51b5\u4e0b\uff0c\u4f60\u53ef\u80fd\u9700\u8981\u6307\u5b9aCUDA\u7248\u672c\u4ee5\u83b7\u5f97GPU\u52a0\u901f\u652f\u6301\u3002\u4f8b\u5982\uff0c\u5b89\u88c5\u652f\u6301CUDA 11.7\u7684Torch\u7248\u672c\uff0c\u4f60\u53ef\u4ee5\u4f7f\u7528\u5982\u4e0b\u547d\u4ee4\uff1a<\/p>\n<\/p>\n
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https:\/\/download.pytorch.org\/whl\/cu117<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u4ee5\u4e0b\u662f\u66f4\u591a\u5173\u4e8e\u5982\u4f55\u5728Python\u4e2d\u4e0b\u8f7dTorch\u7684\u8be6\u7ec6\u6307\u5357\u3002<\/p>\n<\/p>\n
\u4e00\u3001PIP\u5b89\u88c5Torch<\/h3>\n<\/p>\n
1\u3001\u786e\u4fdd\u73af\u5883\u51c6\u5907<\/h4>\n<\/p>\n
\u5728\u5b89\u88c5Torch\u4e4b\u524d\uff0c\u786e\u4fdd\u4f60\u7684\u8ba1\u7b97\u673a\u4e0a\u5df2\u7ecf\u6b63\u786e\u5b89\u88c5\u4e86Python\u548cpip\u3002\u53ef\u4ee5\u901a\u8fc7\u5728\u7ec8\u7aef\u4e2d\u8fd0\u884c\u4ee5\u4e0b\u547d\u4ee4\u6765\u9a8c\u8bc1\uff1a<\/p>\n<\/p>\n
python --version<\/p>\n
pip --version<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u5982\u679c\u4f60\u770b\u5230Python\u548cpip\u7684\u7248\u672c\u53f7\uff0c\u90a3\u4e48\u8bf4\u660e\u5b89\u88c5\u6b63\u786e\u3002\u5982\u679c\u6ca1\u6709\uff0c\u8bf7\u8bbf\u95eePython\u5b98\u65b9\u7f51\u7ad9\u8fdb\u884c\u5b89\u88c5\u3002<\/p>\n<\/p>\n
2\u3001\u4f7f\u7528PIP\u5b89\u88c5<\/h4>\n<\/p>\n
\u5982\u679c\u51c6\u5907\u597d\u4e86\u73af\u5883\uff0c\u63a5\u4e0b\u6765\u53ef\u4ee5\u4f7f\u7528pip\u6765\u5b89\u88c5Torch\u3002\u57fa\u672c\u7684\u5b89\u88c5\u547d\u4ee4\u5982\u4e0b\uff1a<\/p>\n<\/p>\n
pip install torch<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u8fd9\u4e2a\u547d\u4ee4\u4f1a\u5b89\u88c5Torch\u7684CPU\u7248\u672c\u3002\u5982\u679c\u4f60\u5e0c\u671b\u4f7f\u7528GPU\u6765\u52a0\u901f\u8ba1\u7b97\uff0c\u90a3\u4e48\u9700\u8981\u5b89\u88c5\u652f\u6301CUDA\u7684\u7248\u672c\u3002<\/p>\n<\/p>\n
3\u3001\u5b89\u88c5GPU\u7248\u672c<\/h4>\n<\/p>\n
\u5982\u679c\u4f60\u7684\u8ba1\u7b97\u673a\u6709NVIDIA GPU\uff0c\u5e76\u4e14\u5df2\u7ecf\u5b89\u88c5\u4e86CUDA\u548ccuDNN\uff0c\u4f60\u53ef\u4ee5\u5b89\u88c5\u652f\u6301CUDA\u7684Torch\u7248\u672c\u3002\u9996\u5148\uff0c\u8bbf\u95eePyTorch\u5b98\u7f51\u7684\u201cGet Started\u201d\u9875\u9762\uff0c\u6839\u636e\u4f60\u7684\u64cd\u4f5c\u7cfb\u7edf\u3001\u5305\u7ba1\u7406\u5668\u3001\u8bed\u8a00\u9009\u62e9\u4ee5\u53caCUDA\u7248\u672c\uff0c\u751f\u6210\u76f8\u5e94\u7684\u5b89\u88c5\u547d\u4ee4\u3002\u4f8b\u5982\uff1a<\/p>\n<\/p>\n
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https:\/\/download.pytorch.org\/whl\/cu117<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u8fd9\u4e2a\u547d\u4ee4\u5c06\u5b89\u88c5\u652f\u6301CUDA 11.7\u7684Torch\u7248\u672c\u3002<\/p>\n<\/p>\n
\u4e8c\u3001\u4f7f\u7528Conda\u5b89\u88c5Torch<\/h3>\n<\/p>\n
1\u3001\u5b89\u88c5Anaconda\u6216Miniconda<\/h4>\n<\/p>\n
Conda\u662fAnaconda\u53d1\u884c\u7248\u7684\u4e00\u90e8\u5206\uff0c\u4e5f\u53ef\u4ee5\u5355\u72ec\u5b89\u88c5Miniconda\u3002\u786e\u4fdd\u4f60\u5df2\u7ecf\u5b89\u88c5\u4e86\u5176\u4e2d\u4e4b\u4e00\u3002<\/p>\n<\/p>\n
2\u3001\u521b\u5efa\u865a\u62df\u73af\u5883<\/h4>\n<\/p>\n
\u4e3a\u4e86\u907f\u514d\u5305\u4e4b\u95f4\u7684\u51b2\u7a81\uff0c\u5efa\u8bae\u5728\u65b0\u7684Conda\u73af\u5883\u4e2d\u5b89\u88c5Torch\u3002\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u521b\u5efa\u5e76\u6fc0\u6d3b\u73af\u5883\uff1a<\/p>\n<\/p>\n
conda create --name myenv python=3.8<\/p>\n
conda activate myenv<\/p>\n
<\/code><\/pre>\n<\/p>\n
3\u3001\u5b89\u88c5Torch<\/h4>\n<\/p>\n
\u4f7f\u7528Conda\u5b89\u88c5Torch\u4e5f\u975e\u5e38\u7b80\u5355\uff0c\u53ef\u4ee5\u5728\u7ec8\u7aef\u4e2d\u8fd0\u884c\u4ee5\u4e0b\u547d\u4ee4\uff1a<\/p>\n<\/p>\n
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u8fd9\u4e2a\u547d\u4ee4\u4f1a\u81ea\u52a8\u9009\u62e9\u5408\u9002\u7684Torch\u7248\u672c\u5e76\u5b89\u88c5\u6240\u6709\u4f9d\u8d56\u9879\u3002<\/p>\n<\/p>\n
\u4e09\u3001\u4ece\u6e90\u7801\u7f16\u8bd1\u5b89\u88c5<\/h3>\n<\/p>\n
\u867d\u7136\u4e0d\u5e38\u7528\uff0c\u4f46\u5728\u67d0\u4e9b\u60c5\u51b5\u4e0b\uff0c\u4f60\u53ef\u80fd\u9700\u8981\u4ece\u6e90\u7801\u7f16\u8bd1\u5b89\u88c5Torch\u3002\u8fd9\u901a\u5e38\u9002\u7528\u4e8e\u5f00\u53d1\u4eba\u5458\u6216\u9700\u8981\u81ea\u5b9a\u4e49Torch\u7684\u9ad8\u7ea7\u7528\u6237\u3002<\/p>\n<\/p>\n
1\u3001\u51c6\u5907\u6784\u5efa\u73af\u5883<\/h4>\n<\/p>\n
\u9996\u5148\uff0c\u786e\u4fdd\u7cfb\u7edf\u4e0a\u5b89\u88c5\u4e86Git\u3001CMake\u3001Python\u548c\u5176\u4ed6\u5fc5\u8981\u7684\u6784\u5efa\u5de5\u5177\u3002\u4f60\u53ef\u80fd\u8fd8\u9700\u8981\u5b89\u88c5CUDA\u548ccuDNN\u3002<\/p>\n<\/p>\n
2\u3001\u514b\u9686Torch\u6e90\u7801<\/h4>\n<\/p>\n
\u4f7f\u7528Git\u514b\u9686Torch\u7684\u6e90\u7801\u4ed3\u5e93\uff1a<\/p>\n<\/p>\n
git clone --recursive https:\/\/github.com\/pytorch\/pytorch<\/p>\n
cd pytorch<\/p>\n
<\/code><\/pre>\n<\/p>\n
3\u3001\u7f16\u8bd1\u548c\u5b89\u88c5<\/h4>\n<\/p>\n
\u5728\u6e90\u7801\u76ee\u5f55\u4e2d\uff0c\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u7f16\u8bd1\u5e76\u5b89\u88c5Torch\uff1a<\/p>\n<\/p>\n
python setup.py install<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u8fd9\u4e00\u6b65\u53ef\u80fd\u9700\u8981\u4e00\u4e9b\u65f6\u95f4\uff0c\u5e76\u4e14\u9700\u8981\u786e\u4fdd\u6240\u6709\u7684\u4f9d\u8d56\u9879\u90fd\u5df2\u6b63\u786e\u5b89\u88c5\u3002<\/p>\n<\/p>\n
\u56db\u3001\u9a8c\u8bc1\u5b89\u88c5<\/h3>\n<\/p>\n
\u65e0\u8bba\u901a\u8fc7\u54ea\u79cd\u65b9\u6cd5\u5b89\u88c5Torch\uff0c\u90fd\u53ef\u4ee5\u901a\u8fc7\u4ee5\u4e0bPython\u4ee3\u7801\u6765\u9a8c\u8bc1\u5b89\u88c5\u662f\u5426\u6210\u529f\uff1a<\/p>\n<\/p>\n
import torch<\/p>\n
print(torch.__version__)<\/p>\n
print(torch.cuda.is_avAI<\/a>lable())<\/p>\n
<\/code><\/pre>\n<\/p>\n
\u8fd9\u6bb5\u4ee3\u7801\u5c06\u6253\u5370\u51faTorch\u7684\u7248\u672c\u53f7\uff0c\u5e76\u68c0\u67e5CUDA\u662f\u5426\u53ef\u7528\u3002<\/p>\n<\/p>\n
\u4e94\u3001\u5e38\u89c1\u95ee\u9898\u53ca\u89e3\u51b3\u65b9\u6848<\/h3>\n<\/p>\n
1\u3001\u5b89\u88c5\u7f13\u6162\u6216\u5931\u8d25<\/h4>\n<\/p>\n
\u5982\u679c\u5b89\u88c5\u8fc7\u7a0b\u7f13\u6162\u6216\u5931\u8d25\uff0c\u53ef\u80fd\u662f\u7f51\u7edc\u95ee\u9898\u6216\u4f9d\u8d56\u9879\u51b2\u7a81\u5bfc\u81f4\u7684\u3002\u53ef\u4ee5\u5c1d\u8bd5\u4f7f\u7528\u56fd\u5185\u955c\u50cf\u6e90\uff0c\u6216\u8005\u624b\u52a8\u5b89\u88c5\u6240\u9700\u7684\u4f9d\u8d56\u9879\u3002<\/p>\n<\/p>\n
2\u3001CUDA\u7248\u672c\u4e0d\u5339\u914d<\/h4>\n<\/p>\n
\u5b89\u88c5\u652f\u6301CUDA\u7684Torch\u7248\u672c\u65f6\uff0c\u786e\u4fdd\u4f60\u7684CUDA\u548ccuDNN\u7248\u672c\u4e0e\u5b89\u88c5\u547d\u4ee4\u4e2d\u7684\u7248\u672c\u5339\u914d\uff0c\u5426\u5219\u53ef\u80fd\u5bfc\u81f4GPU\u4e0d\u53ef\u7528\u3002<\/p>\n<\/p>\n
3\u3001\u7248\u672c\u517c\u5bb9\u6027\u95ee\u9898<\/h4>\n<\/p>\n
\u6709\u65f6Torch\u7684\u65b0\u7248\u672c\u53ef\u80fd\u4e0e\u5176\u4ed6\u5e93\uff08\u5982NumPy\u3001Pandas\u7b49\uff09\u4e0d\u517c\u5bb9\uff0c\u5bfc\u81f4\u4ee3\u7801\u8fd0\u884c\u9519\u8bef\u3002\u5728\u8fd9\u79cd\u60c5\u51b5\u4e0b\uff0c\u5c1d\u8bd5\u964d\u7ea7Torch\u6216\u5176\u4ed6\u5e93\u7684\u7248\u672c\u4ee5\u89e3\u51b3\u95ee\u9898\u3002<\/p>\n<\/p>\n
\u516d\u3001\u603b\u7ed3<\/h3>\n<\/p>\n
\u5728Python\u4e2d\u4e0b\u8f7d\u548c\u5b89\u88c5Torch\u7684\u65b9\u6cd5\u6709\u591a\u79cd\uff0c\u6700\u5e38\u7528\u7684\u662f\u901a\u8fc7pip\u6216Conda\u8fdb\u884c\u5b89\u88c5\u3002\u65e0\u8bba\u4f7f\u7528\u54ea\u79cd\u65b9\u6cd5\uff0c\u90fd\u9700\u8981\u786e\u4fdd\u73af\u5883\u7684\u6b63\u786e\u914d\u7f6e\uff0c\u4ee5\u53caCUDA\u548ccuDNN\u7248\u672c\u7684\u5339\u914d\u3002\u901a\u8fc7\u4ee5\u4e0a\u6b65\u9aa4\uff0c\u4f60\u5e94\u8be5\u80fd\u591f\u987a\u5229\u5730\u5728\u4f60\u7684Python\u73af\u5883\u4e2d\u5b89\u88c5\u5e76\u4f7f\u7528Torch\u3002\u5e0c\u671b\u8fd9\u7bc7\u6307\u5357\u80fd\u5e2e\u52a9\u4f60\u66f4\u597d\u5730\u7406\u89e3\u548c\u89e3\u51b3\u5728\u5b89\u88c5Torch\u8fc7\u7a0b\u4e2d\u9047\u5230\u7684\u95ee\u9898\u3002<\/p>\n<\/p>\n
\u76f8\u5173\u95ee\u7b54FAQs\uff1a<\/strong><\/h2>\n
 \u5982\u4f55\u5728Python\u4e2d\u5b89\u88c5Torch\uff1f<\/strong>
\u8981\u5728Python\u73af\u5883\u4e2d\u5b89\u88c5Torch\uff0c\u60a8\u53ef\u4ee5\u4f7f\u7528pip\u547d\u4ee4\u3002\u6253\u5f00\u7ec8\u7aef\u6216\u547d\u4ee4\u63d0\u793a\u7b26\uff0c\u8f93\u5165\u4ee5\u4e0b\u547d\u4ee4\uff1a<\/p>\n
pip install torch torchvision torchaudio\n<\/code><\/pre>\n
\u786e\u4fdd\u60a8\u7684Python\u7248\u672c\u4e0eTorch\u517c\u5bb9\uff0c\u901a\u5e38\u9700\u8981Python 3.6\u53ca\u4ee5\u4e0a\u7248\u672c\u3002\u5b89\u88c5\u5b8c\u6210\u540e\uff0c\u60a8\u53ef\u4ee5\u5728Python\u811a\u672c\u4e2d\u901a\u8fc7import torch<\/code>\u6765\u9a8c\u8bc1\u5b89\u88c5\u662f\u5426\u6210\u529f\u3002<\/p>\n
\u5728Windows\u548cLinux\u4e0a\u5b89\u88c5Torch\u7684\u6b65\u9aa4\u6709\u4ec0\u4e48\u4e0d\u540c\uff1f<\/strong>
\u5728Windows\u548cLinux\u4e0a\u5b89\u88c5Torch\u7684\u57fa\u672c\u547d\u4ee4\u76f8\u540c\uff0c\u4f46\u53ef\u80fd\u9700\u8981\u6ce8\u610f\u4e00\u4e9b\u7cfb\u7edf\u7279\u5b9a\u7684\u4f9d\u8d56\u9879\u3002\u5bf9\u4e8eWindows\u7528\u6237\uff0c\u786e\u4fdd\u5df2\u5b89\u88c5Visual Studio\u7684C++\u6784\u5efa\u5de5\u5177\u3002\u6b64\u5916\uff0cLinux\u7528\u6237\u53ef\u80fd\u9700\u8981\u5728\u5b89\u88c5\u524d\u66f4\u65b0\u5305\u7ba1\u7406\u5668\uff0c\u5e76\u786e\u4fdd\u5b89\u88c5\u4e86gcc\u548cg++\u7f16\u8bd1\u5668\u3002<\/p>\n
\u5982\u4f55\u5728\u7279\u5b9a\u7248\u672c\u7684Python\u4e2d\u5b89\u88c5Torch\uff1f<\/strong>
\u5982\u679c\u60a8\u6709\u591a\u4e2aPython\u7248\u672c\u5e76\u5e0c\u671b\u5728\u7279\u5b9a\u7248\u672c\u4e2d\u5b89\u88c5Torch\uff0c\u53ef\u4ee5\u6307\u5b9apip\u7684\u8def\u5f84\u3002\u4f8b\u5982\uff0c\u5982\u679c\u60a8\u60f3\u5728Python3.8\u4e2d\u5b89\u88c5Torch\uff0c\u53ef\u4ee5\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\uff1a<\/p>\n
\/path\/to\/python3.8 -m pip install torch torchvision torchaudio\n<\/code><\/pre>\n
\u786e\u4fdd\u66ff\u6362\/path\/to\/python3.8<\/code>\u4e3a\u60a8\u5b9e\u9645\u7684Python3.8\u8def\u5f84\u3002\u8fd9\u79cd\u65b9\u6cd5\u53ef\u4ee5\u786e\u4fddTorch\u5b89\u88c5\u5728\u60a8\u9009\u62e9\u7684Python\u73af\u5883\u4e2d\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"\u5728Python\u4e2d\u4e0b\u8f7dTorch\u7684\u65b9\u6cd5\u6709\u591a\u79cd\uff0c\u53ef\u4ee5\u901a\u8fc7pip\u547d\u4ee4\u3001conda\u547d\u4ee4\u6216\u8005\u76f4\u63a5\u4ece\u6e90\u7801\u7f16\u8bd1\u5b89\u88c5\u3002pip\u547d […]","protected":false},"author":3,"featured_media":960045,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"_acf_changed":false,"footnotes":""},"categories":[37],"tags":[],"acf":[],"_links":{"self":[{"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/posts\/960039"}],"collection":[{"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/users\/3"}],"replies":[{"embeddable":true,"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/comments?post=960039"}],"version-history":[{"count":"1","href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/posts\/960039\/revisions"}],"predecessor-version":[{"id":960046,"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/posts\/960039\/revisions\/960046"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/media\/960045"}],"wp:attachment":[{"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/media?parent=960039"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/categories?post=960039"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/docs.pingcode.com\/wp-json\/wp\/v2\/tags?post=960039"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}